Skip to content

Commit db552c5

Browse files
authored
Merge pull request #104 from code-for-venezuela/luis/dashboaord-app
Luis/dashboard app
2 parents a6f5838 + 4218384 commit db552c5

40 files changed

+5577
-2836
lines changed

data/processed/huggingface/service_training_dataset.csv

Lines changed: 2400 additions & 2401 deletions
Large diffs are not rendered by default.

experiments_samples/service_classification_experiment.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
training_args={
88
"per_device_train_batch_size" : 15,
99
"per_device_eval_batch_size" : 15, # change batch size if you're having memory problems
10-
"num_train_epochs" : 3,
10+
"num_train_epochs" : 50,
1111
"warmup_steps" : 1000,
1212
"save_steps" : 1000,
1313
"save_total_limit" : 1,
@@ -17,6 +17,7 @@
1717
"eval_accumulation_steps" : 1,
1818
"learning_rate" : 5e-07,
1919
"adafactor" : True,
20+
"metric_for_best_model" : "accuracy"
2021
},
2122
columns=['title'],
2223
train_dataset_name="service_training_dataset.csv", # note that this is the same experiment as relevance classification but using different parameters

experiments_samples/test_relevance_classifier.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,5 +27,5 @@
2727

2828
)
2929

30-
exp = ClassifierExperiment.from_branch_and_experiment("samples", "classifier")
30+
exp = ClassifierExperiment.from_branch_and_experiment("samples", "relevance_classifier")
3131
exp.run_experiment(args)

0 commit comments

Comments
 (0)