From 4066201ad36c2153a288b477f1aaf96f113ced0d Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Thu, 29 May 2025 14:20:49 -0600
Subject: [PATCH] Fix

---
 frigate/util/classification.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/frigate/util/classification.py b/frigate/util/classification.py
index 2b81678d9..e281a0535 100644
--- a/frigate/util/classification.py
+++ b/frigate/util/classification.py
@@ -71,7 +71,7 @@ def train_classification_model(model_dir: str) -> bool:
     # create training set
     datagen = ImageDataGenerator(rescale=1.0 / 255, validation_split=0.2)
     train_gen = datagen.flow_from_directory(
-        "train",
+        dataset_dir,
         target_size=(224, 224),
         batch_size=BATCH_SIZE,
         class_mode="categorical",