File size: 271 Bytes
445afd8
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
{
  "model_type": "polyencoder",
  "encoder_name": "bert-base-uncased",
  "param_count": 109494528,
  "num_steps": 1000,
  "best_step": 950,
  "batch_size": 2,
  "learning_rate": 2e-05,
  "train_time_s": 359.68,
  "precision": 0.9463,
  "recall": 0.9677,
  "f1": 0.9569
}