From ee1a359f4f8cae96e2dbbe68cbe80a7d25dd60e1 Mon Sep 17 00:00:00 2001 From: Rahul Madaan Date: Thu, 1 Oct 2020 14:29:01 +0530 Subject: [PATCH] Update Model_LGBM.py --- Model_LGBM.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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')