Skip to content

ml-labs

Collectors

sunkusun9/ml-labs

Collectors¶

`mllabs.collector.Collector` ¶

`has(node)` ¶

`has_node(node)` ¶

`save()` ¶

`load(path)` `classmethod` ¶

`mllabs.collector.MetricCollector` ¶

Bases: Collector

`get_metric(node)` ¶

`get_metrics(nodes=None)` ¶

`get_metrics_agg(nodes=None, inner_fold=True, outer_fold=True, include_std=False)` ¶

`mllabs.collector.ProbToLabel` ¶

`mllabs.collector.StackingCollector` ¶

Bases: Collector

`get_dataset(nodes=None, include_target=True)` ¶

`mllabs.collector.ModelAttrCollector` ¶

Bases: Collector

`get_attr(node, idx=None)` ¶

`get_attrs(nodes=None)` ¶

`get_attrs_agg(node, agg_inner=True, agg_outer=True)` ¶

`mllabs.collector.SHAPCollector` ¶

Bases: Collector

`get_feature_importance(node, idx)` ¶

`get_feature_importance_agg(node, agg_inner='mean', agg_outer='mean')` ¶

`mllabs.collector.OutputCollector` ¶

Bases: Collector

`get_output(node, outer_idx, inner_idx)` ¶

`get_outputs(node)` ¶

`mllabs.collector.ProcessCollector` ¶

Bases: Collector

`get_output(nodes=None, agg='mean')` ¶