Basic System (#31)

Co-authored-by: Tim Lorsbach <tim@lorsba.ch>
Reviewed-on: enviPath/enviPy#31

envipath/settings.py

@@ -15,6 +15,7 @@ from pathlib import Path
 from dotenv import load_dotenv
 from envipy_plugins import Classifier, Property, Descriptor
 from sklearn.ensemble import RandomForestClassifier
+from sklearn.tree import DecisionTreeClassifier
 
 # Build paths inside the project like this: BASE_DIR / 'subdir'.
 BASE_DIR = Path(__file__).resolve().parent.parent
@@ -256,14 +257,29 @@ CELERY_RESULT_BACKEND = 'redis://localhost:6379/1'
 CELERY_ACCEPT_CONTENT = ['json']
 CELERY_TASK_SERIALIZER = 'json'
 
-DEFAULT_MODELS_PARAMS = {
-    'base_clf': RandomForestClassifier(n_estimators=100,
-                                       max_features='log2',
-                                       random_state=42,
-                                       criterion='entropy',
-                                       ccp_alpha=0.0,
-                                       max_depth=3,
-                                       min_samples_leaf=1),
+DEFAULT_RF_MODEL_PARAMS = {
+    'base_clf': RandomForestClassifier(
+        n_estimators=100,
+        max_features='log2',
+        random_state=42,
+        criterion='entropy',
+        ccp_alpha=0.0,
+        max_depth=3,
+        min_samples_leaf=1
+    ),
+    'num_chains': 10,
+}
+
+DEFAULT_DT_MODEL_PARAMS = {
+    'base_clf': DecisionTreeClassifier(
+        criterion='entropy',
+        max_depth=3,
+        min_samples_split=5,
+        min_samples_leaf=5,
+        max_features='sqrt',
+        class_weight='balanced',
+        random_state=42
+    ),
     'num_chains': 10,
 }