This repository contains a classification model built using TensorFlow. The model is trained on a dataset with 11 features to classify data into one of 5 categories. The implementation includes data preprocessing, model training, evaluation, and suggestions for further improvement.
- Preprocessed dataset to remove invalid columns and handle missing values.
- Multi-class classification using a TensorFlow Sequential model.
- Training, validation, and evaluation metrics for model performance.
- The dataset contains:
- 1986 samples for testing.
- 11 features per sample.
- 5 output classes, represented in a one-hot encoded format.
- Final Training Accuracy: 83.18%
- Final Training Loss: 0.6030
- Test Accuracy: 82.28%
- Test Loss: 0.6186
This project is licensed under the MIT License. Feel free to use and modify it.