增加网格搜索超参数功能。

2025-11-11 18:03:50 +08:00
parent f0468e729b
commit b4b2b18ccd
5 changed files with 526 additions and 0 deletions
@@ -35,6 +35,7 @@ class TaskConfig:
    name: str
    train_dir: str
    val_dir: str
+    test_dir: str  # 测试集目录（用于网格搜索评估）
    embedding_dim: int
    batch_size: int
    lr: float
@@ -48,6 +49,7 @@ TASKS = {
        name='DishClassification',
        train_dir=os.path.join(settings.BASE_DIR, 'dataset', 'DishClassification', 'train'),
        val_dir=os.path.join(settings.BASE_DIR, 'dataset', 'DishClassification', 'val'),
+        test_dir=os.path.join(settings.BASE_DIR, 'dataset', 'DishClassification', 'test'),
        embedding_dim=512,
        batch_size=32,
        lr=5e-4,
@@ -61,6 +63,7 @@ TASKS = {
        name='WholeIngredientRecognition',
        train_dir=os.path.join(settings.BASE_DIR, 'dataset', 'WholeIngredientRecognition', 'train'),
        val_dir=os.path.join(settings.BASE_DIR, 'dataset', 'WholeIngredientRecognition', 'val'),
+        test_dir=os.path.join(settings.BASE_DIR, 'dataset', 'WholeIngredientRecognition', 'test'),
        embedding_dim=512,
        batch_size=64,
        lr=8e-4,
@@ -72,6 +75,7 @@ TASKS = {
        name='ProcessedIngredientRecognition',
        train_dir=os.path.join(settings.BASE_DIR, 'dataset', 'ProcessedIngredientRecognition', 'train'),
        val_dir=os.path.join(settings.BASE_DIR, 'dataset', 'ProcessedIngredientRecognition', 'val'),
+        test_dir=os.path.join(settings.BASE_DIR, 'dataset', 'ProcessedIngredientRecognition', 'test'),
        embedding_dim=512,
        batch_size=32,
        lr=5e-4,