Przewidywanie CTR z użyciem uczenia maszynowego w Pythonie
Kevin Huo
Instructor

| is_student | loan | |
|---|---|---|
| middle_aged | 1 | |
| youth | no | 0 |
| youth | yes | 1 |
clf = DecisionTreeClassifier()Podobnie jak regresja logistyczna, drzewo decyzyjne używa clf.fit(X_train, y_train) do trenowania i clf.predict(X_test) do przewidywania etykiet:
array([0, 1, 1, ..., 1, 0, 1])
clf.predict_proba(X_test) do uzyskania prawdopodobieństw:
array([0.2, 0.8], [0.4, 0.6] ..., [0.1, 0.9] [0.3, 0.7]])
Przykład losowego podziału na dane treningowe i testowe, gdzie dane testowe stanowią 30% próby: train_test_split(X, y, test_size = .3, random_state = 0)

Y_score = clf.predict_proba(X_test)
fpr, tpr, thresholds = roc_curve(Y_test, Y_score[:, 1])
roc_curve() — dane wejściowe: tablice testowa i wynikowaroc_auc = auc(fpr, tpr)
auc() — dane wejściowe: tablice fałszywych i prawdziwych wyników pozytywnych
Jeśli model jest dokładny, a CTR niski, warto ponownie ocenić przekaz reklamy i docelową grupę odbiorców
Przewidywanie CTR z użyciem uczenia maszynowego w Pythonie