Hugging Face's logo

leonvanbokhorst
/
topic-drift-detector

PyTorch
English
topic-drift
conversation-analysis
attention
Eval Results
Model card Files Community
leonvanbokhorst commited on
Commit
d4952c7
·
verified ·
1 Parent(s): 57353d0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -104,12 +104,12 @@ print(f"Topic drift score: {drift_scores.item():.4f}")
104
  ```
105
 
106
  ## Training Details
107
- - Dataset: [leonvanbokhorst/topic-drift](https://huggingface.co/datasets/leonvanbokhorst/topic-drift)
108
  - Window size: 8 turns
109
  - Batch size: 32
110
  - Learning rate: 0.0001
111
  - Early stopping patience: 10
112
- - Total epochs: 37 (early stopped)
113
  - Training framework: PyTorch
114
  - Base embeddings: BAAI/bge-m3
115
 
 
104
  ```
105
 
106
  ## Training Details
107
+ - Dataset: [leonvanbokhorst/topic-drift-v2](https://huggingface.co/datasets/leonvanbokhorst/topic-drift-v2)
108
  - Window size: 8 turns
109
  - Batch size: 32
110
  - Learning rate: 0.0001
111
  - Early stopping patience: 10
112
+ - Total epochs: 70 (early stopped)
113
  - Training framework: PyTorch
114
  - Base embeddings: BAAI/bge-m3
115