2024-01-11 23:20:16 +01:00
|
|
|
This repository contains several scripts integral to the development, training, and deployment of a machine learning model.
|
|
|
|
|
|
|
|
|
|
**classifiers_test.py**
|
|
|
|
|
|
|
|
|
|
*Purpose:* This script is used for testing various classifiers.
|
|
|
|
|
|
|
|
|
|
*Functionality:* It conducts a series of tests on different classifiers to evaluate their performance and suitability for our model.
|
|
|
|
|
|
|
|
|
|
**model_creation.py**
|
|
|
|
|
|
|
|
|
|
*Purpose:* Responsible for creating and training the machine learning model.
|
|
|
|
|
|
|
|
|
|
*Functionality:* Reads input data, trains the model and saves it to trained_model.pkl.
|
|
|
|
|
|
|
|
|
|
**model_inference.py**
|
|
|
|
|
|
|
|
|
|
*Purpose:* Utilizes the trained model to make predictions on new data.
|
|
|
|
|
|
|
|
|
|
*Functionality:* Loads the model from trained_model.pkl, makes predictions on new data and outputs a binary table indicating collapse predictions.
|
