Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Topic 3. Decision tree regressor, MSE #382

Closed
lalimpiev opened this issue Oct 17, 2018 · 1 comment
Closed

Topic 3. Decision tree regressor, MSE #382

lalimpiev opened this issue Oct 17, 2018 · 1 comment

Comments

@lalimpiev
Copy link

В примере по DecisionTreeRegressor неправильный расчет MSE в названии графика:
plt.title("Decision tree regressor, MSE = %.2f" % np.sum((y_test - reg_tree_pred) ** 2))
Нужно ещё поделить на количество наблюдений, предлагаю поправить так:
plt.title("Decision tree regressor, MSE = %.4f" % (np.sum((y_test - reg_tree_pred) ** 2) / n_test))

Файл:
https://github.com/Yorko/mlcourse.ai/blob/master/jupyter_english/topic03_decision_trees_kNN/topic3_decision_trees_kNN.ipynb
И в русской версии аналогично:
https://github.com/Yorko/mlcourse.ai/blob/master/jupyter_russian/topic03_decision_trees_knn/topic3_trees_knn.ipynb

@Yorko
Copy link
Owner

Yorko commented Oct 17, 2018

Thanks, fixed

@Yorko Yorko closed this as completed Oct 17, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants