models = {
    "unweighted": LogisticRegression(max_iter=1000),
    "weighted": LogisticRegression(
        class_weight=weight_map,
        max_iter=1000,
    ),
}