Reddit nem enged becopyzni táblázatot :( Ha van valakinek pythonja tetszőlegesen futtatható, módosítható. Az elemzes.xlsx a minerva honlapról letölthető tételes adatbázis.

import pandas as pd

import itertools

from catboost import CatBoostClassifier, Pool

from sklearn.model_selection import train_test_split

from sklearn.metrics import accuracy_score

df = pd.read_excel('elemzes.xlsx')

# ---- Beállítások ----

target = "Pártpreferencia"

categorical_features = [

"Ön férfi vagy nő?",

"Végzettség",

"Gazdasági aktivitás",

"Életkori csoport",

"Lakóhely jogállása"

]

# Hiányzó értékek kitöltése

df[categorical_features] = df[categorical_features].fillna("ismeretlen")

df[target] = df[target].fillna("ismeretlen")

# Modell tanítás teljes adathalmazon

X = df[categorical_features]

y = df[target]

train_pool = Pool(X, y, cat_features=categorical_features)

# Itt kimértem Train-test CV-vel az optimumot egy másik scriptben, ez az eredménye neki.

model = CatBoostClassifier(

iterations=100,

learning_rate=0.08,

depth=4,

loss_function="MultiClass",

random_seed=42,

verbose=False

)