GitLab wurde erfolgreich aktualisiert. Durch regelmäßige Updates bleibt das THM GitLab sicher. Danke für Ihre Geduld.

Commit f5412e58 authored by Jens Plüddemann's avatar Jens Plüddemann

added clustering

parent ebdfe4d7
from sklearn.cluster import KMeans
import numpy as np
class Clustering:
def __init__(self):
# Erstelle Testdaten, eine Liste an Listen
X = np.array([
[1, 2],
[0, 1],
[4, 2],
[5, 4],
[6, 1],
[1, 4]
])
# Erstellen des Clustering Verfahrens und das hinzufügen der Daten
cluster = KMeans(n_clusters=2, random_state=0).fit(X)
# Schreiben der Ergebnisse auf die Konsole
print(f"Lables: {cluster.labels_}")
print(f"Prediction: {cluster.predict([[0, 0], [5, 3]])}")
print(f"Cluster centers: {cluster.cluster_centers_}")
if __name__ == '__main__':
data = Clustering()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment