| 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'])) | |