From eca972066a011b1438da98b1c58c3d75b4bd8716 Mon Sep 17 00:00:00 2001
From: Klaudia <klamar7@st.amu.edu.pl>
Date: Thu, 11 May 2023 18:40:03 +0200
Subject: [PATCH] fix

---
 create-dataset.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/create-dataset.py b/create-dataset.py
index 64744c0..132e88f 100644
--- a/create-dataset.py
+++ b/create-dataset.py
@@ -6,6 +6,7 @@ import train
 CUTOFF = int(os.environ['CUTOFF'])
 adults = pd.read_csv('adult.csv')
 adults = adults.dropna()
+adults = adults.drop(adults.columns[[1, 3, 4, 5, 6, 7, 8, 9, 13, 14]], axis=1)
 
 adults = adults.sample(CUTOFF)