no neural network

removed from MLJAR and AutoGluon
zuliani99 · Sep 13, 2021 · 5fe6a15 · 5fe6a15
1 parent 2fefa1f
commit 5fe6a15
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/app/algorithms/auto_gluon.py b/app/algorithms/auto_gluon.py
@@ -47,7 +47,7 @@ def autogluon(df, task, options, time_start):
       time_limit=options['time']*60,
       presets=['best_quality'],
       #auto_stack=True
-      #hyperparameters=hyperparameters    # -> Questo aggiunge le NN 
+      hyperparameters=hyperparameters    # -> Comment this line to add te Neural Network
     )
 
     predictor.fit_summary()

diff --git a/app/algorithms/mljar_supervised.py b/app/algorithms/mljar_supervised.py
@@ -41,7 +41,7 @@ def do_mljar(df, options, task, time_start):
     else: ml_task = 'binary_classification'
   else: ml_task=task
 
-  automl = AutoML(mode="Compete", total_time_limit=60*options['time'], ml_task=ml_task)
+  automl = AutoML(mode="Compete", total_time_limit=60*options['time'], ml_task=ml_task, algorithms=["Baseline", "Linear", "Decision Tree", "Random Forest", "Extra Trees", "LightGBM", "Xgboost", "CatBoost"])
 
   automl.fit(X_train, y_train)
   y_pred = automl.predict_all(X_test)