evalml.pipelines.ETMulticlassClassificationPipeline

class evalml.pipelines.ETMulticlassClassificationPipeline(parameters, random_state=0)

Extra Trees Pipeline for multiclass classification.

name = 'Extra Trees Multiclass Classification Pipeline'

custom_name = 'Extra Trees Multiclass Classification Pipeline'

summary = 'Extra Trees Classifier w/ One Hot Encoder + Simple Imputer'

component_graph = ['One Hot Encoder', 'Simple Imputer', 'Extra Trees Classifier']

problem_type = 'multiclass'

model_family = 'extra_trees'

hyperparameters = {'Extra Trees Classifier': {'max_depth': Integer(low=4, high=10, prior='uniform', transform='identity'), 'max_features': ['auto', 'sqrt', 'log2'], 'n_estimators': Integer(low=10, high=1000, prior='uniform', transform='identity')}, 'One Hot Encoder': {}, 'Simple Imputer': {'impute_strategy': ['mean', 'median', 'most_frequent']}}

custom_hyperparameters = None

default_parameters = {'Extra Trees Classifier': {'max_depth': 6, 'max_features': 'auto', 'min_samples_split': 2, 'min_weight_fraction_leaf': 0.0, 'n_estimators': 100, 'n_jobs': -1}, 'One Hot Encoder': {'categories': None, 'drop': None, 'handle_missing': 'error', 'handle_unknown': 'ignore', 'top_n': 10}, 'Simple Imputer': {'fill_value': None, 'impute_strategy': 'most_frequent'}}

Instance attributes

feature_importance	Return importance associated with each feature.
parameters	Returns parameter dictionary for this pipeline

Methods:

__init__	Machine learning pipeline made out of transformers and a estimator.
clone	Constructs a new pipeline with the same parameters and components.
describe	Outputs pipeline details including component parameters
fit	Build a model
get_component	Returns component by name
graph	Generate an image representing the pipeline graph
graph_feature_importance	Generate a bar graph of the pipeline’s feature importance
load	Loads pipeline at file path
predict	Make predictions using selected features.
predict_proba	Make probability estimates for labels.
save	Saves pipeline at file path
score	Evaluate model performance on objectives