Networks Module

class bamt.networks.big_brave_bn.BigBraveBN(n_nearest=5, threshold=0.3, proximity_metric='MI')[source]

Bases: object

set_possible_edges_by_brave(df)[source]

Returns list of possible edges for structure learning

Parameters:

  • df (DataFrame) – data

  • proximity_matrix (DataFrame) – might be generated by get_mutual_info_score_matrix() function, correlation etc.

  • n_nearest (int, optional) – Number of Nearest neighbors, hyperparameter. Defaults to 5.

  • threshold (float, optional) – Threshold for edge candidates to be passed to possible edges, threshold. Defaults to 0.3 [0;1].

Returns:

list of possible edges

Return type:

Possible edges