Randonm forest python

Random forest python

from sklearn.ensemble import RandomForestClassifier

y = train_data["Survived"]

features = ["Pclass", "Sex", "SibSp", "Parch","Fare","Age"]
X = pd.get_dummies(train_data[features])
X_test = pd.get_dummies(test_data[features])

model = RandomForestClassifier(n_estimators=100, max_depth=5, random_state=1)
model.fit(X, y)
predictions = model.predict(X_test)

Source: grepper.com

Similar codes

1 vote179 views1 answer
Stringbuilder python
Python
1 vote179 views1 answer
Use python type hint for multiple return values
Python
0 vote216 views1 answer
Torch.max
Python
0 vote170 views1 answer
Install easygui conda
Python
30 votes168 views6 answers
Try and except in python
Python
1 vote199 views2 answers
Convert timestamp to period pandas
Python
1 vote139 views1 answer
Get time taken to run python
Python
3 votes165 views1 answer
Flask give port number
Python
0 vote158 views1 answer
Extract n grams from text python
Python
11 votes157 views2 answers
Python functions with input
Python