Update app.py
Browse files
app.py
CHANGED
|
@@ -38,7 +38,9 @@ print(label_counts)
|
|
| 38 |
# Labels
|
| 39 |
label_encoder = LabelEncoder()
|
| 40 |
df['label_encoded'] = label_encoder.fit_transform(df['label'])
|
|
|
|
| 41 |
num_classes = len(label_encoder.classes_)
|
|
|
|
| 42 |
|
| 43 |
# Train testsplit
|
| 44 |
X_train, X_test, y_train, y_test = train_test_split(
|
|
|
|
| 38 |
# Labels
|
| 39 |
label_encoder = LabelEncoder()
|
| 40 |
df['label_encoded'] = label_encoder.fit_transform(df['label'])
|
| 41 |
+
print(label_encoder)
|
| 42 |
num_classes = len(label_encoder.classes_)
|
| 43 |
+
print(num_classes)
|
| 44 |
|
| 45 |
# Train testsplit
|
| 46 |
X_train, X_test, y_train, y_test = train_test_split(
|