Classifier ouput features are 4 in Model, then why this printed as 1?
shivanandmn opened this issue · comments
Shivanand commented
Abhishek Kumar Mishra commented
Hi @shivanandmn, thank you for pointing that out. The prompt output is from my previous model that had only 1 neuron in the final layer.
That should be 4 for the current network. However, there is no change in the model performance for this particular dataset. I will re-run the notebook and update it with the correct prompts. Rest assured you can take the notebook as it is and execute.