rulelift/__init__.py at master · aialgorithm/rulelift · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
from .utils import load_example_data, preprocess_data
from .analysis import (
    VariableAnalyzer,
    analyze_rules,
    analyze_rule_correlation,
    get_user_rule_matrix,
    calculate_strategy_gain
)
from .mining import (
    SingleFeatureRuleMiner,
    MultiFeatureRuleMiner,
    DecisionTreeRuleExtractor,
    XGBoostRuleMiner,
    TreeRuleExtractor
)
from .visualization import (
    RuleVisualizer,
    plot_rule_comparison,
    plot_rule_distribution,
    plot_lift_precision_scatter,
    plot_decision_tree,
    plot_rule_network,
    plot_heatmap,
    generate_rule_report
)

__version__ = '1.3.0'
__all__ = [
    # Utils
    'load_example_data',
    'preprocess_data',

    # Analysis
    'VariableAnalyzer',
    'analyze_rules',
    'analyze_rule_correlation',
    'get_user_rule_matrix',
    'calculate_strategy_gain',

    # Mining
    'SingleFeatureRuleMiner',
    'MultiFeatureRuleMiner',
    'DecisionTreeRuleExtractor',
    'XGBoostRuleMiner',  # Deprecated: 请使用 TreeRuleExtractor(algorithm='gbdt')
    'TreeRuleExtractor',

    # Visualization
    'RuleVisualizer',
    'plot_rule_comparison',
    'plot_rule_distribution',
    'plot_lift_precision_scatter',
    'plot_decision_tree',
    'plot_rule_network',
    'plot_heatmap',
    'generate_rule_report'
]