File size: 322 Bytes
c3c4fed 3a9b94c c3c4fed 3a9b94c c3c4fed 3a9b94c c3c4fed 3a9b94c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | prepare:
test_size: 0.2
random_state: 42
max_samples: 5000 # REDUCED from 5000 to 500 for fast CI verification
train:
model_name: "distilbert-base-uncased"
num_epochs: 2
batch_size: 16
learning_rate: 2e-5
warmup_steps: 100
weight_decay: 0.01
max_length: 256 # max token length per review
|