Traceback (most recent call last):
File "learning_embeddings/network/ethec_experiments.py", line 473, in <module>
ETHEC_train_model(args)
File "learning_embeddings/network/ethec_experiments.py", line 361, in ETHEC_train_model
sampler=None if arguments.class_weights else WeightedResampler(
File "learning_embeddings/network/experiment.py", line 231, in __init__
label_ids = [dataset[ind]['leaf_label'] for ind in range(len(dataset))]
File "learning_embeddings/network/experiment.py", line 231, in <listcomp>
label_ids = [dataset[ind]['leaf_label'] for ind in range(len(dataset))]
File "learning_embeddings/data/db.py", line 3789, in __getitem__
'labels': torch.from_numpy(self.labelmap.get_one_hot(sample['family'], sample['subfamily'], sample['genus'],
File "learning_embeddings/data/db.py", line 1071, in get_one_hot
retval[self.genus_specific_epithet[specific_epithet] + self.levels[0] + self.levels[1] + self.levels[2]] = 1
KeyError: 'mannii'
When running
it crashes with:
What am i doing wrong ?