Pohon Keputusan

Machine Learning untuk Keuangan dengan Python

Nathan George

Data Science Professor

Pohon keputusan

pohon keputusan dasar

Machine Learning untuk Keuangan dengan Python

Pohon keputusan

pohon keputusan dengan target

Machine Learning untuk Keuangan dengan Python

Pemisahan pohon keputusan

pemisahan pohon keputusan kategorikal

Machine Learning untuk Keuangan dengan Python

Pemisahan pohon keputusan

pemisahan pohon keputusan numerik

Machine Learning untuk Keuangan dengan Python

Pohon yang buruk

pemisahan regresi yang buruk

Machine Learning untuk Keuangan dengan Python

Pohon yang baik

pemisahan regresi yang baik

Machine Learning untuk Keuangan dengan Python

Regresi pohon keputusan

prediksi pohon keputusan

Machine Learning untuk Keuangan dengan Python

Pohon regresi

from sklearn.tree import DecisionTreeRegressor

decision_tree = DecisionTreeRegressor(max_depth=5)

decision_tree.fit(train_features, train_targets)
Machine Learning untuk Keuangan dengan Python

Hyperparameter pohon keputusan

kedalaman maksimum pohon

Machine Learning untuk Keuangan dengan Python

Kedalaman maksimum 3

pohon keputusan kedalaman maksimum 3

Machine Learning untuk Keuangan dengan Python

Evaluasi model

print(decision_tree.score(train_features, train_targets))
print(decision_tree.score(test_features, test_targets))

0.6662215501032416
-0.08917300191734268

train_predictions = decision_tree.predict(train_features)
test_predictions = decision_tree.predict(test_features)
plt.scatter(train_predictions, train_targets, label='train')
plt.scatter(test_predictions, test_targets, label='test')
plt.legend()
plt.show()
Machine Learning untuk Keuangan dengan Python

prediksi vs aktual

Machine Learning untuk Keuangan dengan Python

Ayo tanam beberapa pohon!

Machine Learning untuk Keuangan dengan Python

Preparing Video For Download...