intial commit (forked from private repo)

extracted_cells/cell23.py

@@ -0,0 +1,13 @@
+# 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))