Я хочу кластеризовать данные, используя иерархическую кластеризацию, и я хочу, чтобы результатом был тип данных целое число, а не число с плавающей запятой
import pandas as pd
import csv
#import csv
#from nltk.cluster import KMeansClusterer, euclidean_distance
dataset =pd.read_csv('vector.csv')
X = dataset
from sklearn.cluster import AgglomerativeClustering
cluster = AgglomerativeClustering(n_clusters=5, affinity='euclidean', linkage='ward')
cluster.fit_predict(X)
#print(cluster.labels_)
file=open('data_hirarki/hirarki.csv','a')
tulis=csv.writer(file,delimiter='\n',lineterminator='\n')
tulis.writerows([cluster.labels_])
file.close()
как я могу это сделать?
результат как
1.0, 2.0, etc
я хочу 1, 2,3