Commit cc95536f authored by Almouhannad Hafez's avatar Almouhannad Hafez

Use Id col in augmented data

parent e85a9af8
......@@ -197,6 +197,9 @@
"metadata": {},
"outputs": [],
"source": [
"augmented_df = augmented_df.drop(columns=['Unnamed: 0'])\n",
"augmented_df['Id'] = range(len(augmented_df))\n",
"augmented_df = augmented_df[['Id', 'label', 'text']]\n",
"augmented_df.to_csv(CONSTANTS.AUGMENTED_DATASET_PATH, index=False)"
]
},
......@@ -302,7 +305,7 @@
"source": [
"df = augmented_df\n",
"\n",
"df = df.drop(columns=['Unnamed: 0'])\n",
"df = df.drop(columns=['Id'])\n",
"df = df[['text', 'label']]\n",
"\n",
"df.head()"
......
This source diff could not be displayed because it is too large. You can view the blob instead.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment