Built a neural network model to solve a basic image classification problem, on MNIST dataset.
Dataset: MNIST dataset
Steps Involved:
- Import Packages (numpy, tensorflow, matplotlib)
- Load Data (load the data and split the data into train and test)
- One Hot Encoding (labels)
- Data Reshaping
- Data Normalization (Mean and Standard Deviation)
- Creating a model (structure, layers)
- Compiling the model (optimizer, loss, metrics)
- Model fitting
- Evaluating the model (loss , accuracy)
- Predictions
- Plotting (Matplotlib)