簡體   English   中英

保存 model 后得到錯誤的預測

[英]Getting Wrong Predictions after saving the model

我的 Model 的源代碼:-

import numpy as np
import pandas as pd

dataset= pd.read_csv("heart900.csv")
X=dataset.iloc[:, :-1].values
Y=dataset.iloc[:, 13].values

from sklearn.impute import SimpleImputer 
imputer = SimpleImputer(missing_values=np.nan, strategy='mean')

imputer= imputer.fit(X)
X= imputer.transform(X)

from sklearn.model_selection import train_test_split
X_Train, X_Test, Y_Train, Y_Test= train_test_split(X,Y, test_size=0.2, random_state=0)

from sklearn.preprocessing import StandardScaler
sc=StandardScaler()
X_Train=sc.fit_transform(X_Train)
X_Test=sc.fit_transform(X_Test) 

import keras
from tensorflow.python.keras.layers import Dense
from tensorflow.python.keras import Sequential
from tensorflow.python.keras.models import load_model
import tensorflow as tf
import warnings


model=Sequential()
##First Hidden Layer
model.add(Dense(6, input_dim=13, activation='relu'))
##Second Hidden Layer
model.add(Dense(6, activation='relu'))
##Third Hidden Layer
model.add(Dense(6, activation='relu'))
##Output Layer
model.add(Dense(1, activation='sigmoid'))

model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])

model.fit(X_Train, Y_Train, epochs=150, batch_size=10)

new_pred= model.predict(sc.transform(np.array([[62,1,1,120,281,0,0,103,0,1.4,1,1,3]])))

new_pred= (new_pred>0.5)
print(new_pred )

model.save('keras_model.h5')`

這里的 O/P 是:- 假

以及用於訪問已保存的 model 的源代碼:-

from tensorflow.python.keras.models import load_model
import numpy as np


mp = load_model('keras_model.h5')
new_pre = mp.predict((np.array([[62,1,1,120,281,0,0,103,0,1.4,1,1,3]])))
new_pre = (new_pre>0.5)
print(new_pre)

這里的 O/P 是:- True(它應該是 False)

而且我已經嘗試了所有可能的方法來保存 h5 model 但每次預測的結果都是錯誤的。 請幫忙!!!!!

您是否忘記在第二次預測時向輸入數據添加轉換?

mp = load_model('keras_model.h5')
new_pre = mp.predict(sc.transform((np.array([[62,1,1,120,281,0,0,103,0,1.4,1,1,3]]))))
new_pre = (new_pre>0.5)
print(new_pre)

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM