module benchmark.sklearn_helper

Short summary

module pymlbenchmark.benchmark.sklearn_helper

Helpers about scikit-learn.

source on GitHub

Functions

function

truncated documentation

get_nb_skl_base_estimators

Returns the number of scikit-learn BaseEstimator including in a pipeline. The function assumes the pipeline …

Documentation

Helpers about scikit-learn.

source on GitHub

pymlbenchmark.benchmark.sklearn_helper.get_nb_skl_base_estimators(obj, fitted=True)

Returns the number of scikit-learn BaseEstimator including in a pipeline. The function assumes the pipeline is not recursive.

Parameters:
  • obj – object to walk through

  • fitted – count the number of fitted object

Returns:

number of base estimators including this one

source on GitHub