diff --git a/attributes.npy b/attributes.npy
new file mode 100644
index 0000000000000000000000000000000000000000..abf10f77bcf949806c0b6bcb54a49bd4eee2b09b
Binary files /dev/null and b/attributes.npy differ
diff --git a/gen_train_data/gen_train_data.ipynb b/gen_train_data/gen_train_data.ipynb
index 8f6a7e04713d0933e8edad19936da054c9c3670f..5688ba2bcfa0058d5198cf131e8774218b4d1ae0 100644
--- a/gen_train_data/gen_train_data.ipynb
+++ b/gen_train_data/gen_train_data.ipynb
@@ -163,6 +163,16 @@
     "np.save('./data/output/post/X_train_under_post.npy', X_train_under_post)\n",
     "np.save('./data/output/post/y_train_under_post.npy', y_train_under_post)"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Save features\n",
+    "np.save('./data/output/attributes.npy', feat)"
+   ]
   }
  ],
  "metadata": {