import matplotlib.pyplot as plt

from pipe import Pipe

from pipetools import *

from utils import load_x

from wrapper import agglomerative_l_method

# file = './datasets/pendigits/pendigits.tra'

file = './datasets/iris/iris.data'

result = Pipe() \

.connect(stop())

X = load_x(file)

centroids = result['centroids']

x = list(map(lambda x: x[0], X))

y = list(map(lambda x: x[1], X))

cen_x = list(map(lambda x: x[0], centroids))

cen_y = list(map(lambda x: x[1], centroids))

plt.scatter(x, y, c='blue')

plt.scatter(cen_x, cen_y, c='red')

plt.grid(True)

plt.show()