Skip to content

Commit b4b61d2

Browse files
edited it to evaluate on full dataset
1 parent 62573f8 commit b4b61d2

File tree

1 file changed

+2
-11
lines changed

1 file changed

+2
-11
lines changed

tabs/evaluator.py

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -166,18 +166,9 @@ def metrics_callback(metrics_df, processed, total):
166166
# Use full dataset for evaluation
167167

168168
try:
169-
# Filter dataset to only first 10 images for evaluation
170-
if hasattr(dataset, "dataset"):
171-
# Create a shallow copy of the dataset object with only first 10 rows
172-
import copy
173-
dataset_subset = copy.copy(dataset)
174-
dataset_subset.dataset = dataset.dataset.iloc[:100].copy()
175-
else:
176-
st.warning("Dataset object does not have a 'dataset' attribute; using as is.")
177-
dataset_subset = dataset
178-
169+
# Use the full dataset for evaluation
179170
results = model.eval(
180-
dataset=dataset_subset,
171+
dataset=dataset,
181172
split=split,
182173
ontology_translation=ontology_translation_path,
183174
predictions_outdir=predictions_outdir,

0 commit comments

Comments
 (0)