thak123's picture
Update config.py
f71a77c
raw
history blame
420 Bytes
from huggingface_hub import hf_hub_download,snapshot_download
print(snapshot_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier"))
model_name = "FFZG-cleopatra/Croatian-News-Classifier""
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