Sklearn OneVsRestClassifier - Training a Model with an OvR Strategy

Python

 1|  from sklearn.ensemble import RandomForestClassifier
 2|  from sklearn.multiclass import OneVsRestClassifier
 3|  
 4|  # Step 1: Initialise estimator
 5|  estimator = RandomForestClassifier(n_estimators=10,
 6|                                 max_depth=None, 
 7|                                 class_weight='balanced',
 8|                                 n_jobs=-1,
 9|                                 random_state=101)
10|  
11|  # Step 2: Pass estimator to one vs rest classifier and fit
12|  model = OneVsRestClassifier(estimator=estimator).fit(X_train, y_train)
13|  
14|  # Step 3: Use model to make predictions for X_test
15|  y_pred = model.predict(X_test)
detro
Machine learning | Classification
Did you find this snippet useful?

Sign up for free to to add this to your code library

Remove Stop Words from Text in DataFrame Column
Python

Pandas | Nltk | Nlp

142
LightGBM Custom Loss Function
Python

LightGBM | Loss function

127
Tuning XGBoost Hyperparameters with Grid Search
Python

XGBoost | Tuning

122
How to Convert DataFrame Values Into Percentages
Python

Lambda | Percentages | Pipe

115
How to Scale Data Using Standard Scaler But Keep Column Names
Python

Scaler | Standard

86
LightGBM Hyperparameter Tuning with GridSearch
Python

Python | LightGBM | Hyperparameter tuning | Gridsearch

85
Calculating Root Mean Squared Error (RMSE) with Sklearn and Python
Python

Rmse

82
Dynamically Create Columns in Pandas Dataframe
Python

Pandas | Columns

65
How to Train a Catboost Classifier with GridSearch Hyperparameter Tuning
Python

Catboost | Hyperparameter tuning

65
Overwriting a Python Logging File Instead of Appending
Python

Logging

61