Spaces:
Running
Running
Update bert_regression.py
Browse files- bert_regression.py +2 -2
bert_regression.py
CHANGED
@@ -16,9 +16,9 @@ def remove_links(review):
|
|
16 |
# df['Text'] = df['Text'].apply(remove_links)
|
17 |
|
18 |
model = AutoModelForSequenceClassification.from_pretrained(
|
19 |
-
'
|
20 |
tokenizer = AutoTokenizer.from_pretrained(
|
21 |
-
'
|
22 |
|
23 |
def preprocess_function_regr(examples):
|
24 |
return tokenizer(examples["Text"], truncation=True, max_length=64, padding = 'max_length')
|
|
|
16 |
# df['Text'] = df['Text'].apply(remove_links)
|
17 |
|
18 |
model = AutoModelForSequenceClassification.from_pretrained(
|
19 |
+
'bert_regr_other_pretrained', num_labels = 1)
|
20 |
tokenizer = AutoTokenizer.from_pretrained(
|
21 |
+
'bert_regr_other_pretrained')
|
22 |
|
23 |
def preprocess_function_regr(examples):
|
24 |
return tokenizer(examples["Text"], truncation=True, max_length=64, padding = 'max_length')
|