Skip to content

Commit

Permalink
✅ Update examples.
Browse files Browse the repository at this point in the history
  • Loading branch information
PGijsbers committed Sep 17, 2019
1 parent 0e4cba1 commit dec6b5a
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion examples/arff_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
print("Make sure you adjust the file path if not executed from the examples directory.")
file_path = "../tests/data/breast_cancer_{}.arff"

automl = GamaClassifier(max_total_time=180, keep_analysis_log=False, n_jobs=1)
automl = GamaClassifier(max_total_time=180, keep_analysis_log=None, n_jobs=1)
print("Starting `fit` which will take roughly 3 minutes.")
automl.fit_arff(file_path.format('train'))

Expand Down
2 changes: 1 addition & 1 deletion examples/classification_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
X, y = load_breast_cancer(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, stratify=y, random_state=0)

automl = GamaClassifier(max_total_time=180, keep_analysis_log=False, n_jobs=1)
automl = GamaClassifier(max_total_time=180, keep_analysis_log=None, n_jobs=1)
print("Starting `fit` which will take roughly 3 minutes.")
automl.fit(X_train, y_train)

Expand Down
2 changes: 1 addition & 1 deletion examples/regression_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
X, y = load_boston(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=0)

automl = GamaRegressor(max_total_time=180, keep_analysis_log=False, n_jobs=1)
automl = GamaRegressor(max_total_time=180, keep_analysis_log=None, n_jobs=1)
print("Starting `fit` which will take roughly 3 minutes.")
automl.fit(X_train, y_train)

Expand Down

0 comments on commit dec6b5a

Please sign in to comment.