mohli commited on
Commit
774e134
·
verified ·
1 Parent(s): 83131d7

Update KNN/KNNModel.py

Browse files
Files changed (1) hide show
  1. 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("app", "data", "gym_members_exercise_tracking.csv") # Updated path
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("app", "KNN", "optimal_k_plot.png"))
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