Skip to content

Commit

Permalink
Fix docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
roquelopez committed Oct 23, 2023
1 parent ba7d513 commit 4c0740b
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions alpha_automl/automl_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -426,7 +426,8 @@ def __init__(self, output_folder, time_bound=15, metric='f1_score', split_strate
:param score_sorting: The sort used to order the scores. It could be `auto`, `ascending` or `descending`.
`auto` is used for the built-in metrics. For the user-defined metrics, this param must be passed.
:param metric_kwargs: Additional arguments for metric.
:param split_strategy_kwargs: Additional arguments for splitting_strategy. In SemiSupervised case, `n_splits` and `test_size`(test proportion from 0 to 1) can be pass to the splitter.
:param split_strategy_kwargs: Additional arguments for splitting_strategy. In SemiSupervised case, `n_splits`
and `test_size`(test proportion from 0 to 1) can be pass to the splitter.
:param start_mode: The mode to start the multiprocessing library. It could be `auto`, `fork` or `spawn`.
:param verbose: Whether or not to show additional logs.
"""
Expand Down Expand Up @@ -465,4 +466,4 @@ def predict_pipeline(self, X, pipeline_id):
def score_pipeline(self, X, y, pipeline_id):
y = self.label_enconder.transform(y)

return super().score_pipeline(X, y, pipeline_id)
return super().score_pipeline(X, y, pipeline_id)

0 comments on commit 4c0740b

Please sign in to comment.