dfinel commited on
Commit
c32bcca
1 Parent(s): 65bf073

Update bert_regression.py

Browse files
Files changed (1) hide show
  1. 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
- '../topic_magnet/bert_regr_other_pretrained', num_labels = 1)
20
  tokenizer = AutoTokenizer.from_pretrained(
21
- '../topic_magnet/bert_regr_other_pretrained')
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')