no neural network

removed from MLJAR and AutoGluon

Author: zuliani99

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()

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)