Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

fixed datasplitting

This commit is contained in:
Falko Victor Habel 2025-01-26 18:40:19 +01:00
parent 51da6b5aa2
commit e0abdb9d39
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/aiia/data/DataLoader.py
View File

@ -165,7 +165,8 @@ class AIIADataLoader:
if len(self.items) == 0:
return [], []
tasks = [item[1] if len(item) > 1 and hasattr(item, '__getitem__') else None for item in self.items]
tasks = [item[1] for item in self.items if len(item) > 1 and hasattr(item, '__getitem__') and item[1] is not None]
unique_tasks = list(set(tasks)) if tasks.count(None) < len(tasks) else []
train_indices = []