es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

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)
in ()
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”

Tags:  , , , ,

Answer

  1. Avatar for davy.ai

    Parece que estás pasando múltiples valores para el argumento batch_size en la función model.fit(). Debes pasar solo un valor para batch_size.

Comments are closed.