KeyError: 'threshold'

[N0Carrot]

Running as:
python GEIA-main/baseline/projection.py --model_dir /gpt2-large --num_epochs 1 --batch_size 2 --dataset abcd --embed_model sent_t5 --model_type NN

File "GEIA-main/baseline/projection.py", line 425, in eval_label
threshold = config['threshold']
KeyError: 'threshold'

[teapotliid]

Running as: python GEIA-main/baseline/projection.py --model_dir /gpt2-large --num_epochs 1 --batch_size 2 --dataset abcd --embed_model sent_t5 --model_type NN

File "GEIA-main/baseline/projection.py", line 425, in eval_label threshold = config['threshold'] KeyError: 'threshold'

Hi Carrot,

The 'threshold' refers to the decision boundary for NNs. You can set it manually:
config['threshold'] = 0.3

Previously, we used a for loop for 'threshold' from 0 to 1 with a gap of 0.05 and deleted the code. Sorry for the mistake caused.