16 lines
484 B
Python
16 lines
484 B
Python
import joblib
|
|
import numpy as np
|
|
from sklearn.ensemble import RandomForestClassifier
|
|
|
|
# Buat model Random Forest dummy
|
|
model = RandomForestClassifier(n_estimators=10, random_state=42)
|
|
|
|
# Training dengan data dummy (24 fitur untuk histogram RGB 8x8x8)
|
|
X_dummy = np.random.rand(100, 24)
|
|
y_dummy = np.random.choice([0, 1, 2], 100)
|
|
model.fit(X_dummy, y_dummy)
|
|
|
|
# Simpan model
|
|
joblib.dump(model, 'model_tomat.pkl')
|
|
print('Model dummy berhasil dibuat dan disimpan sebagai model_tomat.pkl')
|