Spaces:
Sleeping
Sleeping
feat: Removed create_dummy_model function
Browse files
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:
|