@@ -77,14 +77,15 @@ class MVTecLOCO(AnomalibDataModule):
77
77
Defaults to ``None``.
78
78
augmentations (Transform | None): General augmentations to apply if stage-specific
79
79
augmentations are not provided.
80
- test_split_mode (TestSplitMode): Method to create test set.
80
+ Defaults to ``None``.
81
+ test_split_mode (TestSplitMode | str): Method to create test set.
81
82
Defaults to ``TestSplitMode.FROM_DIR``.
82
- test_split_ratio (float ): Fraction of data to use for testing .
83
- Defaults to ``0.2 ``.
84
- val_split_mode (ValSplitMode ): Method to create validation set .
85
- Defaults to ``ValSplitMode.SAME_AS_TEST ``.
86
- val_split_ratio (float): Fraction of data to use for validation.
87
- Defaults to ``0.5 ``.
83
+ val_split_mode (ValSplitMode | str ): Method to create validation set .
84
+ Defaults to ``ValSplitMode.FROM_DIR ``.
85
+ test_split_ratio (float | None ): Fraction of data to use for testing .
86
+ Defaults to ``None ``.
87
+ val_split_ratio (float | None ): Fraction of data to use for validation.
88
+ Defaults to ``None ``.
88
89
seed (int | None, optional): Seed for reproducibility.
89
90
Defaults to ``None``.
90
91
@@ -130,9 +131,9 @@ def __init__(
130
131
val_augmentations : Transform | None = None ,
131
132
test_augmentations : Transform | None = None ,
132
133
augmentations : Transform | None = None ,
133
- test_split_mode : TestSplitMode | str | None = None ,
134
+ test_split_mode : TestSplitMode | str = TestSplitMode .FROM_DIR ,
135
+ val_split_mode : ValSplitMode | str = ValSplitMode .FROM_DIR ,
134
136
test_split_ratio : float | None = None ,
135
- val_split_mode : ValSplitMode | str | None = None ,
136
137
val_split_ratio : float | None = None ,
137
138
seed : int | None = None ,
138
139
) -> None :
@@ -145,8 +146,8 @@ def __init__(
145
146
test_augmentations = test_augmentations ,
146
147
augmentations = augmentations ,
147
148
test_split_mode = test_split_mode ,
148
- test_split_ratio = test_split_ratio ,
149
149
val_split_mode = val_split_mode ,
150
+ test_split_ratio = test_split_ratio ,
150
151
val_split_ratio = val_split_ratio ,
151
152
seed = seed ,
152
153
)
0 commit comments