I try to do tune cross-validation with keras model and ASHA ASHAScheduler but i don’t know how to add crosvalidation. Does anybody now same tutorial or example?
rliaw
June 1, 2021, 8:30pm
2
Maybe you can use a KerasClassifier (their scikit-learn interface), and then use tune-sklearn?
"""
An example training a Keras model, performing
grid search using TuneGridSearchCV.
"""
from keras.datasets import mnist
from keras.layers import Dense, Activation, Dropout
from keras.models import Sequential
from keras.utils import np_utils
from keras.wrappers.scikit_learn import KerasClassifier
from tune_sklearn import TuneGridSearchCV
nb_classes = 10
(X_train, y_train), (X_test, y_test) = mnist.load_data()
X_train = X_train[:500]
y_train = y_train[:500]
X_test = X_test[:100]
y_test = y_test[:100]
X_train = X_train.reshape(X_train.shape[0], 784)
show original
@rliaw Thank Yoy for the answer. I will try it.