Module connectome.models.pipeline_RF

function to create and compute a random forest classification or regression model

Functions

def run_random_forest(X_train, y_train, classification: bool = True, verbose=0, **kwargs)

Function that fits a random forest model. See also https://bit.ly/3Jj6fta and https://bit.ly/3CNSUXr for further arguments that can be specified.

Examples:

>>> rf = run_random_forest(X_train = X_train,
                              y_train = y_train,
                              classification = True)   

Args

X_train

The training dataset

y_train

The true labels

classification

classification or regression task

verbose

amount of verbosity, default = 0

Returns

Returns fitted model