The dataset is from https://www.kaggle.com/datasets/ritesaluja/bank-note-authentication-uci-data Kaggle challenge. The dataset has been created by applying wavelet transform to photographs of banknotes and then calculating the features mentioned in the dataset by deriving the histogram of the dataset. For the sake of this project, we limit the analysis of the data; We just derive a good enough model using random forest classification (which already got ~98% accuracy). Then, we pickle the model and load into the Flask backend. We have created a quick UI with flasgger. Then, we dockerized it.
-
Notifications
You must be signed in to change notification settings - Fork 0
ktjayamanna/banknotes_forgery_classification
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
The purpose of this project is to demo how a small ML model can be deployed to the web using Flask and docker
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published