Spaces:
Sleeping
Sleeping
Update trainDistilBERT.py
Browse files- trainDistilBERT.py +1 -3
trainDistilBERT.py
CHANGED
@@ -10,13 +10,11 @@ from transformers.modeling_outputs import SequenceClassifierOutput
|
|
10 |
from datasets import load_dataset, Dataset
|
11 |
import pandas as pd
|
12 |
from sklearn.metrics import accuracy_score, precision_recall_fscore_support, confusion_matrix, roc_auc_score
|
13 |
-
import wandb
|
14 |
import huggingface_hub
|
15 |
import os
|
16 |
import random
|
17 |
import numpy as np
|
18 |
|
19 |
-
WANDB_MODE="offline"
|
20 |
|
21 |
config = {
|
22 |
"mdl": "distilbert-base-uncased",
|
@@ -33,7 +31,7 @@ config = {
|
|
33 |
"epsilon": 1e-8
|
34 |
}
|
35 |
|
36 |
-
wandb.init(project="MarkIII_ACARIS", entity="simtoonia", config=config)
|
37 |
|
38 |
|
39 |
def lockSeed(seed):
|
|
|
10 |
from datasets import load_dataset, Dataset
|
11 |
import pandas as pd
|
12 |
from sklearn.metrics import accuracy_score, precision_recall_fscore_support, confusion_matrix, roc_auc_score
|
|
|
13 |
import huggingface_hub
|
14 |
import os
|
15 |
import random
|
16 |
import numpy as np
|
17 |
|
|
|
18 |
|
19 |
config = {
|
20 |
"mdl": "distilbert-base-uncased",
|
|
|
31 |
"epsilon": 1e-8
|
32 |
}
|
33 |
|
34 |
+
#wandb.init(project="MarkIII_ACARIS", entity="simtoonia", config=config)
|
35 |
|
36 |
|
37 |
def lockSeed(seed):
|