fix #260

hackingmaterials · Nov 8, 2019 · 26bd6c7 · 26bd6c7
1 parent 15d10e3
commit 26bd6c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/automatminer/automl/adaptors.py b/automatminer/automl/adaptors.py
@@ -109,7 +109,7 @@ def fit(self, df, target, **fit_kwargs):
         # Determine learning type based on whether classification or regression
         self.mode = regression_or_classification(df[target])
 
-        mltype_str = "Classifier" if self.mode == AMM_CLF_NAME else "Regressor"
+        # mltype_str = "Classifier" if self.mode == AMM_CLF_NAME else "Regressor"
         # self.tpot_kwargs["template"] = self.tpot_kwargs.get(
         #     "template", "Selector-Transformer-{}".format(mltype_str)
         # )