Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"
|