eml.tree.reader package

Submodules

eml.tree.reader.sklearn_reader module

eml.tree.reader.sklearn_reader.read_sklearn_tree(tree)

Import decision tree model from sklearn

Casts decision tree into custom representation, available at eml.tree.describe.DTNode

Parameters:tree (sklearn.tree.BaseDecisionTree) – Trained decision tree Returns:Decision tree – Decision tree with custom representation Return type:eml.tree.describe.DTNode

Module contents