Commit cc95536f authored by Almouhannad Hafez's avatar Almouhannad Hafez

Use Id col in augmented data

parent e85a9af8
...@@ -197,6 +197,9 @@ ...@@ -197,6 +197,9 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "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)" "augmented_df.to_csv(CONSTANTS.AUGMENTED_DATASET_PATH, index=False)"
] ]
}, },
...@@ -302,7 +305,7 @@ ...@@ -302,7 +305,7 @@
"source": [ "source": [
"df = augmented_df\n", "df = augmented_df\n",
"\n", "\n",
"df = df.drop(columns=['Unnamed: 0'])\n", "df = df.drop(columns=['Id'])\n",
"df = df[['text', 'label']]\n", "df = df[['text', 'label']]\n",
"\n", "\n",
"df.head()" "df.head()"
......
This diff is collapsed.
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