初試tensorflow(二)——第一段代碼

按照《程序員的AI書》中的示例,照貓畫虎,可運(yùn)行出錯(cuò)。問題出在model.fit,說是參數(shù)類型不對(duì)。示例中輸入的是兩個(gè)列表:
x_array = [1, 2, 3, 10, 2000, -2, -10, -100, -5,-20]
y= [1.0,1.0,1.0,1.0,1.0,0.0,0.0,0.0,0.0,0.0]
檢查了一下,應(yīng)該是軟件的版本不一致造成的,目前的版本model.fit可以接收numpy.ndarray類型。
將上面的代碼修為:
x_array = np.array( [1, 2, 3, 10, 2000, -2, -10, -100, -5,-20])
y= np.array([1.0,1.0,1.0,1.0,1.0,0.0,0.0,0.0,0.0,0.0])
程序可以通過了。
完整的代碼如下:

import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
import numpy as np

model = Sequential()
model.add(Dense(units=8,activation='relu',input_dim=1))
model.add(Dense(units=1,activation='sigmoid'))
model.compile(loss='mean_squared_error',optimizer='sgd')
x_array = np.array( [1, 2, 3, 10, 2000, -2, -10, -100, -5,-20])
y= np.array([1.0,1.0,1.0,1.0,1.0,0.0,0.0,0.0,0.0,0.0])
model.fit(x_array,y,epochs=10,batch_size=4)

test_x=np.array([30,40,-20,-60])
test_y=model.predict(test_x)

for i in range(0,len(test_x)):
    print('input {} => predict:{}'.format(test_x[i],test_y[i]))
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容