Update KNN/KNNModel.py
Browse files- KNN/KNNModel.py +2 -2
KNN/KNNModel.py
CHANGED
@@ -10,7 +10,7 @@ import matplotlib.pyplot as plt
|
|
10 |
class KNNModel:
|
11 |
def __init__(self):
|
12 |
self.target_column = "Fat_Percentage"
|
13 |
-
self.data_path = os.path.join("
|
14 |
self.data = None
|
15 |
self.selected_features = None
|
16 |
self.label_encoders = {}
|
@@ -69,7 +69,7 @@ class KNNModel:
|
|
69 |
plt.ylabel("Cross-Validated R^2")
|
70 |
plt.title("R^2 vs. k")
|
71 |
plt.grid()
|
72 |
-
plt.savefig(os.path.join("
|
73 |
plt.close()
|
74 |
|
75 |
return self.optimal_k
|
|
|
10 |
class KNNModel:
|
11 |
def __init__(self):
|
12 |
self.target_column = "Fat_Percentage"
|
13 |
+
self.data_path = os.path.join("data", "gym_members_exercise_tracking.csv") # Updated path
|
14 |
self.data = None
|
15 |
self.selected_features = None
|
16 |
self.label_encoders = {}
|
|
|
69 |
plt.ylabel("Cross-Validated R^2")
|
70 |
plt.title("R^2 vs. k")
|
71 |
plt.grid()
|
72 |
+
plt.savefig(os.path.join("KNN", "optimal_k_plot.png"))
|
73 |
plt.close()
|
74 |
|
75 |
return self.optimal_k
|