Wafer Classification

This application trains ML models on wafer sensor data and classifies if the wafers are Good or Bad. The dataset contains wafer names and 590 columns of different sensor values for each wafer. Datasets for training and prediction are provided separately as a set of csv files. A 'schema' file is used to verify the data format. Logging is performed frequently and log files are stored in the Prediction and Training specific log folders. The code for this application can be found here.

Application Access: The application is deployed on HEROKU and can be accessed via POSTMAN

• Prediction End Point: https://fathomless-brook-01494.herokuapp.com/predict
• Request Type: POST
• Header: {'Content-Type':'application/json'}
• Body: {"filepath":"Prediction_Batch_files"}

Libraries Used:

• App framework - flask, wsgiref
• DB operations - sqlite3
• Data processing and ML - numpy, pandas, matplotlib, sklearn, xgboost, kneed, pickle
• General operations - os, shutil, csv, json, re, datetime, time