def add_histogram_text_support():
add_node_to_str(Histogram, histogram_to_str)
add_str_to_spn(
"histogram", histogram_tree_to_spn, """
histogram: "Histogram(" FNAME "|" list ";" list ")" """,
None)
def add_piecewise_text_support():
add_node_to_str(PiecewiseLinear, piecewise_to_str)
add_str_to_spn(
"pwl", piecewise_tree_to_spn, """
pwl: "PiecewiseLinear(" FNAME "|" list ";" list ")" """,
None)
def add_histogram_text_support():
add_node_to_str(Conditional, conditional_to_str)
add_str_to_spn(
"conditional", conditional_to_spn, """
conditional: "conditional(" FNAME "|" list ";" list ")" """,
None)
def add_typed_leaves_text_support():
add_parametric_text_support()
add_node_to_str(TypeMixture, type_mixture_to_str)
add_node_to_str(TypeMixtureUnconstrained, type_mixture_uncon_to_str)
add_str_to_spn("typemixture", type_mixture_tree_to_spn,
"""typemixture: "TypeMixture(" [PARAMNAME]"," [DECIMAL "*" node ("+" DECIMAL "*" node)*] ")" """,
TypeMixture)
add_str_to_spn("typemixtureunconstrained", type_mixture_uncon_tree_to_spn,
"""typemixtureunconstrained: "TypeMixtureUnconstrained(" [PARAMNAME]"," [DECIMAL "*" node ("+" DECIMAL "*" node)*] ")" """,
TypeMixtureUnconstrained)
def add_parametric_text_support():
for c in Parametric.__subclasses__():
add_node_to_str(c, parametric_to_str)
for c in Parametric.__subclasses__():
name = c.__name__
add_str_to_spn(
"parametric_" + name.lower(), parametric_tree_to_spn,
"""%s: "%s(" FNAME "|" [PARAMNAME "=" NUMBERS (";" PARAMNAME "=" NUMBERS )*] ")" """
% ("parametric_" + name.lower(), name), c)
add_str_to_spn(
"categorical", categorical_tree_to_spn,
"""categorical: "Categorical(" FNAME "|" [ PARAMNAME "=" list ] ")" """,
Categorical)
def add_parametric_text_support():
for c in Parametric.__subclasses__():
if(c.__name__ == 'MultivariateGaussian'):
add_node_to_str(MultivariateGaussian, mvg_to_str)
else:
add_node_to_str(c, parametric_to_str)
for c in Parametric.__subclasses__():
if(c.__name__ == 'MultivariateGaussian'):
name = c.__name__
add_str_to_spn(
"parametric_" +
name.lower(),
MVG_tree_to_spn,
"""%s: "%s(" FNAME "|" [ PARAMNAME "=" list ] ")" """ %
("parametric_" +
name.lower(),
name),
c,
)
else:
name = c.__name__
add_str_to_spn(
"parametric_" +
name.lower(),
parametric_tree_to_spn,
"""%s: "%s(" FNAME "|" [PARAMNAME "=" NUMBERS (";" PARAMNAME "=" NUMBERS )*] ")" """ %
("parametric_" +
name.lower(),
name),
c,
)
add_str_to_spn(
"categorical",
categorical_tree_to_spn,
"""categorical: "Categorical(" FNAME "|" [ PARAMNAME "=" list ] ")" """,
Categorical,
)
def add_cltree_text_support():
add_node_to_str(CLTree, cltree_to_str)
