diff --git a/Model_LGBM.py b/Model_LGBM.py index 9bd1d7a..8d713a6 100644 --- a/Model_LGBM.py +++ b/Model_LGBM.py @@ -1,3 +1,5 @@ +# All the imports + import numpy as np import pandas as pd import matplotlib.pyplot as plt @@ -210,4 +212,4 @@ def plotRoc(testLabels, y_pred_probabilities, label): print(confusion_matrix(Y_test, YPred)) print(np.mean(Y_test==YPred)*100) -plotRoc(Y_test,YProb,'ROC_Curve') \ No newline at end of file +plotRoc(Y_test,YProb,'ROC_Curve')