diff --git a/evaluation.py b/evaluation.py index 12ac135..a17a806 100644 --- a/evaluation.py +++ b/evaluation.py @@ -5,12 +5,13 @@ import matplotlib.pyplot as plt from keras import backend as K from sklearn.metrics import mean_squared_error -model = keras.models.load_model('suicide_model.h5') +# model = keras.models.load_model('suicide_model.h5') -# train = pd.read_csv('train.csv') -# test = pd.read_csv('test.csv') -# validate = pd.read_csv('validate.csv') +train = pd.read_csv('train.csv') +test = pd.read_csv('test.csv') +validate = pd.read_csv('validate.csv') +print(train) # # podziaƂ train set # X_train = train.loc[:, train.columns != 'suicides_no'] # y_train = train[['suicides_no']]