Untitled
unknown
plain_text
2 years ago
648 B
6
Indexable
best_score = 0
best_est = 0
for est in range(10,101,10):
model = RandomForestClassifier(random_state=1234,n_estimators=est)
model.fit(features_train,target_train)
score = model.score(features_valid,target_valid)
if score>best_score:
best_score = score
best_est = est
print("Accuracy of the best on the validation set (n_estimators = {})".format(best_est,best_score))
RFmodel = RandomForestClassifier(random_state=1234,n_estimators=100)
RFmodel.fit(features_train,target_train)
RFpredic_valid = RFmodel.predict(features_valid)
RFprob_valid = RFmodel.predict_proba(features_valid)
RFprob_one_valid = RFprob_valid[:,1]Editor is loading...
Leave a Comment