from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import load_iris
import numpy as np
# 1. Data Iris (Bunga)
data = load_iris()
X, y = data.data, data.target
# Target: 0=Setosa, 1=Versicolor, 2=Virginica
# 2. Model Random Forest
# n_estimators=100 (100 Pohon)
rf = RandomForestClassifier(n_estimators=100, max_depth=3, random_state=42)
# 3. Training
rf.fit(X, y)
# 4. Prediksi
sample_bunga = [[5.1, 3.5, 1.4, 0.2]] # Mirip Setosa
prediksi = rf.predict(sample_bunga)
print("Jenis Bunga:", data.target_names[prediksi[0]])
# 5. Feature Importance (Fitur mana yang paling menentukan?)
imp = rf.feature_importances_
print("Pentingnya Fitur:", imp)
# Output misal: [0.1, 0.05, 0.45, 0.4] -> Petal Length & Width paling penting!