Tensorflow : logits and labels must have the same first dimension

by

The problem is in your target shape and is related to the correct choice of an appropriate loss function. you have 2 possibilities:

1. possibility: if you have 1D integer encoded target, you can use sparse_categorical_crossentropy as loss function

n_class = 3
n_features = 100
n_sample = 1000

X = np.random.randint(0,10, (n_sample,n_features))
y = np.random.randint(0,n_class, n_sample)

inp = Input((n_features,))
x = Dense(128, activation='relu')(inp)
out = Dense(n_class, activation='softmax')(x)

model = Model(inp, out)
model.compile(loss="sparse_categorical_crossentropy",optimizer="adam",metrics=['accuracy'])
history = model.fit(X, y, epochs=3)

2. possibility: if you have one-hot encoded your target in order to have 2D shape (n_samples, n_class), you can use categorical_crossentropy

n_class = 3
n_features = 100
n_sample = 1000

X = np.random.randint(0,10, (n_sample,n_features))
y = pd.get_dummies(np.random.randint(0,n_class, n_sample)).values

inp = Input((n_features,))
x = Dense(128, activation='relu')(inp)
out = Dense(n_class, activation='softmax')(x)

model = Model(inp, out)
model.compile(loss="categorical_crossentropy",optimizer="adam",metrics=['accuracy'])
history = model.fit(X, y, epochs=3)

More Related Contents:

  1. TensorFlow Only running on 1/32 of the Training data provided [duplicate]
  2. Make a custom loss function in keras
  3. Keras not training on entire dataset
  4. How to export Keras .h5 to tensorflow .pb?
  5. Could not load dynamic library ‘cudart64_101.dll’ on tensorflow CPU-only installation
  6. How to apply gradient clipping in TensorFlow?
  7. Keras + Tensorflow and Multiprocessing in Python
  8. Keras replacing input layer
  9. How does Keras 1d convolution layer work with word embeddings – text classification problem? (Filters, kernel size, and all hyperparameter)
  10. Keras Sequential model with multiple inputs
  11. ValueError at /image/ Tensor Tensor(“activation_5/Softmax:0”, shape=(?, 4), dtype=float32) is not an element of this graph
  12. What do I need K.clear_session() and del model for (Keras with Tensorflow-gpu)?
  13. expected ndim=3, found ndim=2
  14. This model has not yet been built error on model.summary()
  15. How to get other metrics in Tensorflow 2.0 (not only accuracy)?
  16. How can I use a pre-trained neural network with grayscale images?
  17. Removing then Inserting a New Middle Layer in a Keras Model
  18. Why can’t I get reproducible results in Keras even though I set the random seeds?
  19. AttributeError: module ‘tensorflow.python.keras.utils.generic_utils’ has no attribute ‘populate_dict_with_module_objects’
  20. Custom loss function in Keras
  21. Save and load model optimizer state
  22. Keras misinterprets training data shape
  23. How do I get the weights of a layer in Keras?
  24. ValueError: Input 0 of layer sequential is incompatible with the layer: : expected min_ndim=4, found ndim=3. Full shape received: [8, 28, 28]
  25. Keras LSTM input dimension setting
  26. ‘Tensor’ object has no attribute ‘lower’
  27. Unexpected keyword argument ‘ragged’ in Keras
  28. ValueError: Output tensors to a Model must be the output of a TensorFlow `Layer`
  29. how to convert numpy to tfrecords and then generate batches?
  30. Invalid Argument Error / Graph Execution Error

Leave a Comment