# Buku "Python untuk GIS", Bab 26: Hutan dalam Film Pendek: Deret Waktu, Perubahan, dan Klasifikasi dengan Skrip
# Kotak perkakas kedua: hutan_mini.py
from hutan_mini import RandomForestClassifier
X = [[0.86, 0.02], [0.84, 0.03], [0.88, 0.02], [0.52, 0.15], [0.48, 0.17], [0.55, 0.14]]
y = [1, 1, 1, 2, 2, 2]          # 1 = hutan, 2 = sawah
m = RandomForestClassifier(n_estimators=10, random_state=1).fit(X, y)
print("kelas dikenal:", m.classes_)
print("prediksi      :", m.predict([[0.85, 0.02], [0.50, 0.16]]))
print("kepentingan   :", m.feature_importances_.round(2))