【发布时间】:2021-01-30 05:35:35
【问题描述】:
在 K-fold 循环内部还是外部创建 Keras 模型对象? 请解释为什么你的答案是正确的。
def model_def():
model = Sequential()
model.add(.... so on....)
model.compile(....so on ....)
return model
案例 1:- 在 K-fold 循环内,因此它正在为每个循环重新创建
for train_index, test_index in kf.split(X,Y):
model = model_def()
model.fit(X[train_index],Y[test_index] ..... so on .....
或者, 案例 2:- 在循环之外,因此所有折叠循环的单个模型对象
model = model_def()
for train_index, test_index in kf.split(X,Y):
model.fit(X[train_index],Y[test_index] ..... so on .....
【问题讨论】:
标签: keras scikit-learn deep-learning cross-validation k-fold