martin-fabbri
/
jigsaw-multilingual-toxic-comment
mirror of https://github.com/martin-fabbri/jigsaw-multilingual-toxic-comment.git


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
            dataset:
  bert_gcp_path: gs://bert_multilingual_public/bert_multi_cased_L-12_H-768_A-12_2/
  sequence_lenght: 128
  train_processed: data/raw/jigsaw-toxic-comment-train-processed-seqlen128.csv
train:
  epochs: 12
  batch_size: 128
  lr:
    start: 0.00001
    max: 0.00005
    min: 0.00001
    rampup_epochs: 5
    sustain_epochs: 0
    exp_decay: 0.8
  saved_model_path: models/x
  train_plot_path: plots/x
evaluate:
  evaluate_metrics_path: metrics/evaluate_metrics.json