zaidmehdi commited on
Commit
07cffcd
1 Parent(s): 402d8e8

remove print

Browse files
Files changed (1) hide show
  1. 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
- print(df_train.columns)
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])