disable categoricals strategy change for neural network (#289)

mljar · Jan 11, 2021 · d984cb5 · d984cb5
1 parent d08f6b4
commit d984cb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/supervised/tuner/mljar_tuner.py b/supervised/tuner/mljar_tuner.py
@@ -506,7 +506,7 @@ def get_categorical_strategy(self, current_models, strategy, total_time_limit):
             if m_type not in [
                 "Xgboost",
                 "LightGBM",
-                "Neural Network",
+                # "Neural Network",
                 # "Random Forest",
                 # "Extra Trees",
             ]: