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. Loggin is performed frequently and log files are stored in the Prediction and Training specific log folders.

Application Access: The application 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