File size: 462 Bytes
c8eed0a
f71a77c
c8eed0a
f286b50
93ab1cf
2dd65a9
 
2de6916
 
 
 
 
 
 
 
93ab1cf
 
56f9eb8
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import os

print(os.listdir())
model_name = "EMBEDDIA/crosloengual-bert" #"FFZG-cleopatra/dummy-model" #"FFZG-cleopatra/Croatian-News-Classifier"

print("model-name:",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

from pip import _internal
print(_internal.main(['list']))