s444439/ium_z444439
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Klaudia 2023-05-11 18:37:18 +02:00
parent a839ae2bb9
commit 2b95806379
2 changed files with 20 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
create-dataset.py
View File

@ -1,6 +1,7 @@
import pandas as pd import pandas as pd
import os import os
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split
import train
CUTOFF = int(os.environ['CUTOFF']) CUTOFF = int(os.environ['CUTOFF'])
adults = pd.read_csv('adult.csv') adults = pd.read_csv('adult.csv')
@ -20,3 +21,5 @@ X_test.to_csv('X_test.csv', index=False)
Y_test.to_csv('Y_test.csv', index=False) Y_test.to_csv('Y_test.csv', index=False)
Y_train.to_csv('Y_train.csv', index=False) Y_train.to_csv('Y_train.csv', index=False)
Y_dev.to_csv('Y_dev.csv', index=False) Y_dev.to_csv('Y_dev.csv', index=False)
train.main()

2
train.py
View File

@ -2,6 +2,8 @@ import pandas as pd
import tensorflow import tensorflow
from keras.applications.densenet import layers from keras.applications.densenet import layers
def main():
train_data_x = pd.read_csv('./X_train.csv') train_data_x = pd.read_csv('./X_train.csv')
adults_train = train_data_x.copy() adults_train = train_data_x.copy()