Python Models.build_model_DT Examples

Example #1

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

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

    # Default and tuned Random Forest model predictions
    RF_model = Models.build_model_RF()
    RF_tuned = Models.build_optimized_RF()
    RF_predictions = RF_model.predict(testData)
    RF_tuned_predictions = RF_tuned.predict(testData)
    # Saving predictions
    csv_saver(predictions=RF_predictions, type="RF")
    csv_saver(predictions=RF_tuned_predictions, type="RF_tuned")

    # Decision model predictions
    SVC_model = Models.build_model_SVC()