diff --git a/examples/decision_tree/main.py b/examples/decision_tree/main.py index d7922e682c4f0c510cc14d463c8ca522d855a878..1b82988a33f5717f934beed0f664cebd47c0a3b5 100644 --- a/examples/decision_tree/main.py +++ b/examples/decision_tree/main.py @@ -18,13 +18,8 @@ data = SyntheticData(shuffle=False) config.X_train, config.X_test, config.y_train, config.y_test = data.generator() -#instantiate model here. -#finalModel = Models() - - testModel = decisionTreemodel(config.X_train, config.X_test, config.y_train, config.y_test, 9) -#testModel1 = logisticRegressionModel(config.X_train, config.X_test, config.y_train, config.y_test, 9) #fit model using fit method on the model class