rifatmon commited on
Commit
8de7be2
·
verified ·
1 Parent(s): b9c890b

feat: Removed create_dummy_model function

Browse files
Files changed (1) hide show
  1. model.py +0 -36
model.py CHANGED
@@ -108,43 +108,7 @@ class WellnessRecommendationModel:
108
 
109
  except Exception as e:
110
  logger.error(f"Error training model: {str(e)}")
111
- self._create_dummy_model()
112
 
113
- def _create_dummy_model(self):
114
- """Create a dummy model for testing purposes when dataset is not available."""
115
- logger.warning("Creating dummy model for testing purposes")
116
-
117
- self.label_encoder = LabelEncoder()
118
- self.label_encoder.classes_ = np.array(self.target_classes)
119
-
120
- preprocessor = self._create_preprocessor()
121
-
122
- self.model_pipeline = Pipeline(steps=[
123
- ('preprocessor', preprocessor),
124
- ('classifier', RandomForestClassifier(
125
- n_estimators=10,
126
- random_state=42
127
- ))
128
- ])
129
-
130
- dummy_data = {
131
- 'Umur': [25, 30, 35, 40, 45],
132
- 'Jenis Kelamin': ['Laki-laki', 'Perempuan', 'Laki-laki', 'Perempuan', 'Laki-laki'],
133
- 'Kota': ['Jakarta', 'Bali', 'Yogyakarta', 'Surabaya', 'Bandung'],
134
- 'Frekuensi': ['Sekali sebulan', 'Setiap tahun', 'Setiap tiga bulan', 'Dua kali setahun', 'Sekali setiap 2 bulan'],
135
- 'Anggaran/Kunjungan (IDR)': [1000000, 2000000, 1500000, 3000000, 2500000],
136
- 'Kemauan Bepergian': ['Domestik', 'Regional', 'Internasional', 'Domestik', 'Regional'],
137
- 'Metode Pemesanan': ['Aplikasi Seluler', 'Pemesanan Email', 'Media Sosial', 'Website', 'Telepon'],
138
- 'Tujuan Utama': ['Relaksasi', 'Kesehatan', 'Spiritual', 'Kecantikan', 'Detoks'],
139
- 'Yang Dicari': ['Kecantikan', 'Kesehatan', 'Relaksasi', 'Spiritual', 'Detoks']
140
- }
141
-
142
- X_dummy = pd.DataFrame(dummy_data)
143
- y_dummy = np.array([0, 1, 2, 3, 4])
144
-
145
- self.model_pipeline.fit(X_dummy, y_dummy)
146
- self.is_trained = True
147
-
148
  def _save_model(self):
149
  """Save the trained model and label encoder."""
150
  try:
 
108
 
109
  except Exception as e:
110
  logger.error(f"Error training model: {str(e)}")
 
111
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
112
  def _save_model(self):
113
  """Save the trained model and label encoder."""
114
  try: