En un modelo de red neuronal CNN intento ajustar mis datos usando fit.model() pero me está mostrando un error.
Aquí,
X_train = el 75% de mis datos de imagen de cáncer, que tienen 3 clases.
Y_train = las imágenes están etiquetadas como [0,0,0,0,0,1,1,1,1,1,1,1,2,2,2,2,2]
X_test = el 25% de las imágenes de mi conjunto de datos de cáncer
resultados = modelo.fit(X_train, Y_train, X_test, validation_split=0.1, batch_size=6, epochs=5,
#
Pero estoy obteniendo estos errores,
¿qué debo pasar a model.fit()?
TypeError Traceback (most recent call last)
5 tf.keras.callbacks.TensorBoard(log_dir=’logs’)]
6
—-> 7 resultados = modelo.fit(X_train, Y_train, X_test, validation_split=0.1, batch_size=6, epochs=5, callbacks=callbacks)
8
9 ####################################
1 frames
/usr/local/lib/python3.7/dist-packages/keras/utils/traceback_utils.py in error_handler(*args, **kwargs)
62 filtered_tb = None
63 try:
—> 64 return fn(*args, **kwargs)
65 except Exception as e: # pylint: disable=broad-except
66 filtered_tb = _process_traceback_frames(e.__traceback__)
TypeError: fit() tiene múltiples valores para el argumento “batch_size”
davy.ai
Parece que estás pasando múltiples valores para el argumento
batch_size
en la funciónmodel.fit()
. Debes pasar solo un valor parabatch_size
.