Python Models.build_model_SGD Examples

Example #1

File: predictions.py Project: augustsemrau/Pandas_Scikit-Learn_Classification_Model-Optimization

    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")

    # Decision model predictions
    KNN_model = Models.build_model_KNN()
    KNN_predictions = KNN_model.predict(testData)
    # Saving predictions
    csv_saver(predictions=KNN_predictions, type="KNN")

    # Decision model predictions
    DT_model = Models.build_model_DT()
    DT_predictions = DT_model.predict(testData)
    # Saving predictions
    csv_saver(predictions=DT_predictions, type="DT")