ketanchaudhary88 commited on
Commit
40660d3
Β·
verified Β·
1 Parent(s): 1e4ae57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_l
56
  # Define the training arguments
57
  training_args = TrainingArguments(
58
  output_dir='./results',
59
- evaluation_strategy='epoch', # Evaluate after each epoch
60
  learning_rate=2e-5,
61
  per_device_train_batch_size=8,
62
  per_device_eval_batch_size=8,
 
56
  # Define the training arguments
57
  training_args = TrainingArguments(
58
  output_dir='./results',
59
+ eval_strategy='epoch', # Evaluate after each epoch
60
  learning_rate=2e-5,
61
  per_device_train_batch_size=8,
62
  per_device_eval_batch_size=8,