13 lines
584 B
Python
13 lines
584 B
Python
# Formatierte Regeln anzeigen
|
|
def format_rules(tree_rules):
|
|
"""Formatiert die Baumregeln mit menschenlesbaren Klassennamen"""
|
|
# Ersetze 'class: 0' durch 'Einkommen ≤ 50K'
|
|
formatted_rules = tree_rules.replace('class: 0', 'Einkommen ≤ 50K')
|
|
# Ersetze 'class: 1' durch 'Einkommen > 50K'
|
|
formatted_rules = formatted_rules.replace('class: 1', 'Einkommen > 50K')
|
|
return formatted_rules
|
|
|
|
# Regeln aus dem Surrogate-Baum extrahieren
|
|
tree_rules = export_text(surrogate_tree, feature_names=X_train.columns.tolist())
|
|
#print(tree_rules)
|
|
print(format_rules(tree_rules)) |