我在拟合部分遇到错误。x_train和y_train是否有问题?
import keras from keras.models import Sequential from keras.layers import Dense model=Sequential() model.add(Dense(input_dim=1500,init="random_uniform",activation='sigmoid',output_dim=1000))#输入层 model.add(Dense(output_dim=100,init="random_uniform",activation='sigmoid'))#隐藏层 model.add(Dense(output_dim=1,init='random_uniform',activation='sigmoid'))#输出层 model.compile(optimizer='adam',loss='binary_crossentropy',metrics=['accuracy'])#adam=批量梯度下降 model.fit(x_train,y_train,epochs=50,batch_size=10)#epochs迭代次数 y_pred = model.predict(x_test)
回答:
正如错误信息所示,您的x_train
应该是一个形状为(1500,)
的向量,因为您在第一层中设置了input_dim=1500
,但看起来您传递的x_train
向量的形状是(1,)
。您需要纠正传递给模型的x_train的形状。
展示您读取数据并将其存储为x_train
的代码。这将有助于找出错误所在。