from data_loader import dataLoader
from models import Models
from data_loader import csv_saver
import xgboost as xgb
if __name__ == '__main__':
# Import test data
testData = dataLoader(test=True, ageNAN='median')
# Initiate models
Models = Models(agenan='median')
# Logistic Regression model predictions
logisticModel = Models.build_model_LR()
logistic_predictions = logisticModel.predict(testData)
# Saving predictions
csv_saver(predictions=logistic_predictions, type="logistic")
# Naive Bayes model predictions
naiveBayes_model = Models.build_model_NB()
naiveBayes_predictions = naiveBayes_model.predict(testData)
# Saving predictions
csv_saver(predictions=naiveBayes_predictions, type="naiveBayes")
# Stochastic Gradient Descent model predictions
SGD_model = Models.build_model_SGD()
SGD_predictions = SGD_model.predict(testData)
# Saving predictions
csv_saver(predictions=SGD_predictions, type="SGD")
