Spaces:
Runtime error
Runtime error
model_name = "" | |
output_path = "" | |
drop_out = 0.3 | |
max_length = 512 | |
epochs = 5 | |
label2id = {'neutral': 0, 'negative': 1, 'positive': 2} | |
id2label = {0: 'neutral', 1: 'negative', 2: 'positive'} | |
output_dir="" | |
batch_size=16 #32 | |
learning_rate=2e-5 |