Update final.py

final.py

@@ -3,6 +3,7 @@ import pandas as pd
 import seaborn as sns
 import matplotlib.pyplot as plt
 import numpy as np
+import sklearn
 import warnings
 from sklearn.linear_model import LogisticRegression
 from sklearn.model_selection import train_test_split