membagi muatan

# Kmeans pernah buat bikin pembagian muatan freezer vaksin

import pandas as pd
import numpy as np
from sklearn.cluster import KMeans
from k_means_constrained import KMeansConstrained

data = pd.read_csv('sulsel.csv')
X = np.array(data)

kmeans = KMeans(
    init="random" 
    n_clusters=8 
    n_init=10 
    max_iter=300 
    random_state=42
)

# clf = KMeansConstrained(
#     n_clusters=4,
#     size_min=20,
#     size_max=21,
#     random_state=0
# )

clf = KMeansConstrained(
    n_clusters=3,
    size_min=15,
    size_max=16,
    random_state=0
)

clf.fit_predict(X)
labels = clf.labels_
data['kmc-lab'] = labels
data.to_csv('jabar-clust.csv')


data.to_csv('sulsel-clust.csv')
data = pd.read_csv('sulsel.csv')
X = np.array(data)
clf = KMeansConstrained(
    n_clusters=5,
    size_min=17,
    size_max=19,
    random_state=0
)
clf.fit_predict(X)
labels = clf.labels_
data['kmc-labs'] = labels
data.to_csv('sulsel-clust.csv')