s434784/ium_434784
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

.
All checks were successful
s434784-evaluation/pipeline/head This commit looks good
Details
s434784-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Maciej Sobkowiak 2021-05-16 22:22:32 +02:00
parent beadfe027d
commit b965a730c5
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
evaluation.py
View File

@ -5,12 +5,13 @@ import matplotlib.pyplot as plt
from keras import backend as K from keras import backend as K
from sklearn.metrics import mean_squared_error 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') train = pd.read_csv('train.csv')
# test = pd.read_csv('test.csv') test = pd.read_csv('test.csv')
# validate = pd.read_csv('validate.csv') validate = pd.read_csv('validate.csv')
print(train)
# # podział train set # # podział train set
# X_train = train.loc[:, train.columns != 'suicides_no'] # X_train = train.loc[:, train.columns != 'suicides_no']
# y_train = train[['suicides_no']] # y_train = train[['suicides_no']]