Update Home.py

Home.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import numpy as np
-import sklearn
 from sklearn.datasets import load_iris
 from sklearn.model_selection import train_test_split
 from sklearn.preprocessing import StandardScaler, OneHotEncoder
@@ -93,7 +92,7 @@ def main():
     y = iris.target.reshape(-1, 1)  # shape (150, 1)
 
     # One-hot encode target
-    encoder = OneHotEncoder(sparse=False)
+    encoder = OneHotEncoder(sparse_output=False)
     y_encoded = encoder.fit_transform(y)  # shape (150, 3)
 
     # Split data
@@ -188,5 +187,5 @@ def main():
         st.write(f"#### Test Loss: {loss:.4f}")
         st.write(f"#### Test Accuracy: {acc:.4f}")
 
-if _name_ == "__main__":
-    main()
+if __name__ == "__main__":
+    main()