Inzynierka_Gwiazdy/machine_learning
2024-01-11 23:20:16 +01:00
..
Lib/site-packages Machine learning 2023-09-20 19:46:58 +02:00
Scripts Machine learning 2023-09-20 19:46:58 +02:00
classifiers_test.py Added scaler 2024-01-11 21:42:37 +01:00
main.py Machine learning 2023-09-20 19:46:58 +02:00
model_creation.py Added model creation script with Naive Bayes (var_smoothing=1e-07) as classifier 2024-01-11 21:42:37 +01:00
model_inference.py Added simple imputation to model inference 2024-01-11 21:42:37 +01:00
pyvenv.cfg Machine learning 2023-09-20 19:46:58 +02:00
README.md Added README 2024-01-11 23:20:16 +01:00
starclusters-global-parameters2.dat Machine learning 2023-09-20 19:46:58 +02:00
trained_model.pkl Added trained model as .pkl file" 2024-01-11 21:42:37 +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.