vastorbit.machine_learning.memmodel.tree.Tree

class vastorbit.machine_learning.memmodel.tree.Tree

Base Class for tree representation.

__init__()

Methods

__init__()
get_attributes()	Returns the model attributes.
plot_tree([pic_path])	Draws the input tree.
predict(X)	Predicts using the BinaryTree model.
predict_proba(X)	Returns the model probabilities.
predict_proba_sql(X)	Returns the SQL code needed to deploy the model probabilities.
predict_sql(X)	Returns the SQL code needed to deploy the model.
set_attributes(**kwargs)	Sets the model attributes.
to_graphviz([feature_names, classes_color, ...])	Returns the code for a Graphviz tree.

Attributes

object_type

Must be overridden in child class