In [7]:
# vektor y definiĊĦe pripadnost klasi
# prva tri elementa pripadaju klasi 1
# poslednja tri elementa pripadaju klasi 2

y = np.array([1, 1, 1, 2, 2, 2])