KeyError: 'threshold'

Open N0Carrot opened this issue 2 years ago • 1 comments

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'

Feb 01 '24 05:02 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'

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.

Feb 05 '24 08:02 teapotliid