Spaces:
Sleeping
Sleeping
remove print
Browse files- src/utils.py +1 -1
src/utils.py
CHANGED
@@ -6,7 +6,7 @@ def convert_df_to_dataset_dict(df_train:pd.DataFrame, df_test:pd.DataFrame)-> Da
|
|
6 |
mapper = {"#2_tweet": "tweet", "#3_country_label": "label"}
|
7 |
df_train = df_train.rename(columns=mapper)
|
8 |
df_test = df_test.rename(columns=mapper)
|
9 |
-
|
10 |
columns_to_keep = ["tweet", "label"]
|
11 |
train_dataset = Dataset.from_pandas(df_train[columns_to_keep])
|
12 |
test_dataset = Dataset.from_pandas(df_test[columns_to_keep])
|
|
|
6 |
mapper = {"#2_tweet": "tweet", "#3_country_label": "label"}
|
7 |
df_train = df_train.rename(columns=mapper)
|
8 |
df_test = df_test.rename(columns=mapper)
|
9 |
+
|
10 |
columns_to_keep = ["tweet", "label"]
|
11 |
train_dataset = Dataset.from_pandas(df_train[columns_to_keep])
|
12 |
test_dataset = Dataset.from_pandas(df_test[columns_to_keep])
|