Challenge: Implementing a Decision Tree
In this challenge, you will use the titanic dataset. It holds information about passengers on the Titanic, including their age, sex, family size, etc. And the task is to predict whether a person survived or not.
1234import pandas as pd df = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/b71ff7ac-3932-41d2-a4d8-060e24b00129/titanic.csv') print(df.head())
To implement the Decision Tree, you can use the DecisionTreeClassifier
from the sklearn
.
Your task is to build a Decision Tree and find the best max_depth
and min_samples_leaf
using grid search.
Swipe to start coding
- Import the
DecisionTreeClassifier
class fromsklearn.tree
. - Assign an instance of
DecisionTreeClassifier
to thedecision_tree
variable. - Create a dictionary for a
GridSearchCV
to run through[1, 2, 3, 4, 5, 6, 7]
values ofmax_depth
and[1, 2, 4, 6]
values ofmin_samples_leaf
. - Create a
GridSearchCV
object and train it.
Рішення
Дякуємо за ваш відгук!
single
Запитати АІ
Запитати АІ
Запитайте про що завгодно або спробуйте одне із запропонованих запитань, щоб почати наш чат
Awesome!
Completion rate improved to 3.57
Challenge: Implementing a Decision Tree
Свайпніть щоб показати меню
In this challenge, you will use the titanic dataset. It holds information about passengers on the Titanic, including their age, sex, family size, etc. And the task is to predict whether a person survived or not.
1234import pandas as pd df = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/b71ff7ac-3932-41d2-a4d8-060e24b00129/titanic.csv') print(df.head())
To implement the Decision Tree, you can use the DecisionTreeClassifier
from the sklearn
.
Your task is to build a Decision Tree and find the best max_depth
and min_samples_leaf
using grid search.
Swipe to start coding
- Import the
DecisionTreeClassifier
class fromsklearn.tree
. - Assign an instance of
DecisionTreeClassifier
to thedecision_tree
variable. - Create a dictionary for a
GridSearchCV
to run through[1, 2, 3, 4, 5, 6, 7]
values ofmax_depth
and[1, 2, 4, 6]
values ofmin_samples_leaf
. - Create a
GridSearchCV
object and train it.
Рішення
Дякуємо за ваш відгук!
Awesome!
Completion rate improved to 3.57single