train model nb

sgoldenlab · Feb 21, 2024 · 3a627a1 · 3a627a1
1 parent 75fb550
commit 3a627a1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/nb/train_models.ipynb b/docs/nb/train_models.ipynb
@@ -5,7 +5,7 @@
    "id": "488098e3",
    "metadata": {},
    "source": [
-    "# TRAIN MODELS I"
+    "# Train models I"
    ]
   },
   {
@@ -67,7 +67,7 @@
    ],
    "source": [
     "###Create an instance of a model trainer based on the settings in the project config\n",
-    "model_trainer = TrainRandomForestClassifier(config_path='/Users/simon/Desktop/envs/simba/troubleshooting/two_black_animals_14bp/project_folder/project_config.ini')"
+    "model_trainer = TrainRandomForestClassifier(config_path=CONFIG_PATH)"
    ]
   },
   {
@@ -152,7 +152,7 @@
    ],
    "source": [
     "###Create an instance of a grid model trainer.\n",
-    "model_trainer = GridSearchRandomForestClassifier(config_path='/Users/simon/Desktop/envs/simba/troubleshooting/two_black_animals_14bp/project_folder/project_config.ini')"
+    "model_trainer = GridSearchRandomForestClassifier(config_path=CONFIG_PATH)"
    ]
   },
   {