# Überprüfen der Verteilung der Zielklasse
plt.figure(figsize=(8, 6))
sns.countplot(x='income', data=df)
plt.title('Verteilung der Einkommensklassen')
plt.xlabel('Einkommen')
plt.ylabel('Anzahl')
plt.savefig('output/Verteilung_Einkommensklassen.png', dpi=300)
plt.show()

# Prozentuale Verteilung berechnen
income_counts = df['income'].value_counts(normalize=True) * 100
print("Prozentuale Verteilung der Einkommensklassen:")
print(income_counts)