We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 09c0ee6 commit e8e6c58Copy full SHA for e8e6c58
medcat/utils/meta_cat/ml_utils.py
@@ -14,7 +14,8 @@
14
from sklearn.metrics import classification_report, precision_recall_fscore_support, confusion_matrix
15
from sklearn.model_selection import train_test_split
16
from sklearn.utils.class_weight import compute_class_weight
17
-from transformers import AdamW, get_linear_schedule_with_warmup
+from transformers import get_linear_schedule_with_warmup
18
+from torch.optim import AdamW
19
20
21
import logging
0 commit comments