【发布时间】:2019-09-17 21:35:46
【问题描述】:
我正在尝试使用 Keras 训练我的模型,并且我正在学习 udemy 的这个在线课程。现在一切正常,但是当我尝试将 ANN 拟合到训练集时,它会出现以下错误。一切正常,但是当我执行最后一行时,它给出了错误。 它应该可以正常工作而不会出现此错误,还是有任何其他方法可以将 ANN 拟合到训练集?
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
dataset = pd.read_csv('Churn_Modelling.csv')
X = dataset.iloc[:, 3:13].values
y = dataset.iloc[:, 13].values
from sklearn.preprocessing import OneHotEncoder, LabelEncoder
labelencoder_X_1 = LabelEncoder()
X[:, 1] = labelencoder_X_1.fit_transform(X[:, 1])
labelencoder_X_2 = LabelEncoder()
X[:, 2] = labelencoder_X_2.fit_transform(X[:, 2])
onehotencoder = OneHotEncoder(categorical_features = [1])
X = onehotencoder.fit_transform(X).toarray()
X = X[:, 1:]
from sklearn.model_selection import train_test_split
X_train , y_train , X_test, y_test = train_test_split(X,y, test_size = 0.2 , random_state = 0)
#convert X_test into a 'numpy' array to acoid valur error for 1D array
X_test = np.reshape(y, (-1,1))
from sklearn.preprocessing import StandardScaler
sc = StandardScaler()
X_train = sc.fit_transform(X_train)
X_test = sc.fit_transform(X_test)
import keras
from keras.models import Sequential
from keras.layers import Dense
#initializing the ANN
classifier = Sequential()
#adding the input layer and the first hidden layer
classifier.add(Dense(units =6, kernel_initializer = 'uniform' , activation = 'relu', input_dim =11 ))
#adding the second layer
classifier.add(Dense(units = 6 , kernel_initializer = 'uniform' , activation = 'relu'))
#adding the output layer
classifier.add(Dense(units = 1 , kernel_initializer = 'uniform' , activation = 'sigmoid'))
#compiling the ANN
classifier.compile(optimizer = 'adam' , loss = 'binary_crossentropy', metrics = ['accuracy'])
# 'optimizer' is the algorithm that u wanna use for the wights adjustments
#fitting the ann to the trainging set
classifier.fit(X_train , y_train , batch_size =10 , epochs = 100 )
【问题讨论】:
-
由于您使用的是 OneHotEncoder,我不确定您的 input_dim == 11。尝试:input_dim = X_train.shape[1]
-
您必须仅在训练数据上拟合 StandardScaler:sc.fit(X_train); X_train = sc.transform(X_train); X_test = sc.transform(X_test)。 labelEncoder 和 OnehotEncoder 同上,您必须适应训练数据并处理新标签。
-
@Amine 它仍然给出同样的错误。