I predict 4 different eye diseases by machine learning.
- Import Needed Modules 2 Define a function to print text in color
- Define a function to plot the number of images in dataset classes 4 Define a function to return information on the dataset
- Read in images and create a dataframe of image paths and class labels
- Define a function that trims a dataset for the max number of class images
- Balance the trainning set using augmentation
- Create train, test and validation generators
- Create a function to show Training Image Samples 10 Create a function to calculate the F1 score metric
- Create the Model
- Create a custom Keras callback to continue or halt training
- Define a function to plot the training data
- Define a function save the training data to a csv file
- Define a function to make predictions on the test set
- Make predictions on test set, create Confusion Matrix and Classification Report
- Define a function to save the trained model 18 Define the run function which runs the classifier 19 Define the code to enable to initiate the run function
- Evaluate Model Performance