ClefChen commited on
Commit
7f99c38
·
verified ·
1 Parent(s): 912edf0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -96,8 +96,8 @@ train_data, test_data= train_test_split(df, test_size=0.15, random_state=42 )
96
  train_data['label'].value_counts().sort_index()
97
  test_data['label'].value_counts().sort_index()
98
  vectorizer.fit(train_data.text)
99
- vectorizer.get_feature_names_out()[: 100]
100
- vectorizer= vectorizer
101
  #########################################################################################################################
102
  if torch.cuda.is_available():
103
  device = "cuda"
 
96
  train_data['label'].value_counts().sort_index()
97
  test_data['label'].value_counts().sort_index()
98
  vectorizer.fit(train_data.text)
99
+ # vectorizer.get_feature_names_out()[: 100]
100
+ # vectorizer= vectorizer
101
  #########################################################################################################################
102
  if torch.cuda.is_available():
103
  device = "cuda"