【发布时间】:2020-02-23 09:11:06
【问题描述】:
我正在编写此代码并不断获得支持的目标类型是:('binary', 'multiclass')。取而代之的是“连续”。无论我尝试什么都会出错。您在我的代码中看到问题了吗?
df = pd.read_csv('drain.csv')
values = df.values
seed = 7
numpy.random.seed(seed)
X = df.iloc[:,:2]
Y = df.iloc[:,2:]
def create_model():
# create model
model = Sequential()
model.add(Dense(12, input_dim=8, activation='relu'))
model.add(Dense(8, activation='relu'))
model.add(Dense(1, activation='sigmoid'))
# Compile model
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
return model
model = KerasClassifier(build_fn=create_model, epochs=10, batch_size=10, verbose=0)
# evaluate using 10-fold cross validation
kfold = StratifiedKFold(n_splits=10, shuffle=True, random_state=seed)
results = cross_val_score(model, X, Y, cv=kfold)
print(results.mean())
【问题讨论】:
-
请粘贴错误。你在这里遗漏了一些东西。最终输出应该是二进制,而不是您收到连续输出。
-
您给出了错误的输出目标。您正在尝试对数据进行分类,分类目标应该是一个热向量,因此请检查存储在 Y 中的元素。在您的情况下,它应该是向量而不是矩阵。它的元素应该是零或一。
标签: python machine-learning keras cross-validation k-fold