In [1]:
import numpy as np
import matplotlib.pyplot as plt
In [2]:
# adat beolvasása file-ból
# space-szel elválasztott oszlopok
data = np.genfromtxt('perc3.txt')
In [3]:
# plot, ábra mérete
plt.figure(figsize=(20, 10), dpi=300)
# scatterplot(x koordináta, y koordináta, color='szín')
# a tömbnél először a "data[:,0]==0"-val kiválasztjuk azokatasorokat,aminek a nulladik eleme
# nulla, azaz a nullás osztályba tartoznak, majd a [:,1] az x koordinátája, a [:,2] pedig
# az y koordinátája
plt.scatter(data[data[:,0]==0][:,1],\
           data[data[:,0]==0][:,2],color='red')
plt.scatter(data[data[:,0]==1][:,1],\
           data[data[:,0]==1][:,2],color='orange')
# kirajzoljuk a plotot, amit eddig összeraktunk
plt.show()
In [4]:
# adat elosztása tanító és teszthalmazra, a feature-ök (most koordináták) és a címkék külön változóba kerülnek
# új sklearnnél sklearn.model_selection-re nem dob figyelmeztetést
from sklearn.cross_validation import train_test_split
train_features, test_features, \
train_labels, test_labels = \
train_test_split(data[:,1:], data[:,0], \
                test_size = 0.33, random_state = 12)
C:\Users\virtual\Anaconda3\lib\site-packages\sklearn\cross_validation.py:44: DeprecationWarning: This module was deprecated in version 0.18 in favor of the model_selection module into which all the refactored classes and functions are moved. Also note that the interface of the new CV iterators are different from that of this module. This module will be removed in 0.20.
  "This module will be removed in 0.20.", DeprecationWarning)
In [5]:
# sima euklideszi távolság
def distance(vect1, vect2):
    return np.linalg.norm(vect1-vect2)
In [6]:
# minden test pontra kiszámoljuk az adott pont távolságát a tanítópontokétól a dist tömbbe
# majd kiválasztjuk a legközelebbi címkéjét, azt jósoljuk
pred_labels=np.ndarray(test_labels.size)
for j in range(test_labels.size):
    dist=np.ndarray(train_labels.size)
    for i in range(train_labels.size):
        dist[i]=distance(test_features[j], \
                         train_features[i])
    pred_labels[j]=train_labels[np.argmin(dist)]
In [7]:
# piros: tanító, 0-ás osztály
# narancs: tanító, 1-es osztály
# kék: jósolt 0-ás osztály
# zöld: jósolt 1-es osztály
plt.figure(figsize=(20, 10), dpi=300)
plt.scatter(train_features[train_labels==0][:,0],\
           train_features[train_labels==0][:,1],\
           color='red')
plt.scatter(train_features[train_labels==1][:,0],\
           train_features[train_labels==1][:,1],\
           color='orange')
plt.scatter(test_features[pred_labels==0][:,0],\
           test_features[pred_labels==0][:,1],\
           color='blue')
plt.scatter(test_features[pred_labels==1][:,0],\
           test_features[pred_labels==1][:,1],\
           color='green')
plt.show()
In [8]:
# rosszul jósolt elemek összeszámolása
mispredict=0
for i in range(test_labels.size):
    if test_labels[i] != pred_labels[i]:
        mispredict += 1
print(mispredict)
1
In [9]:
# adat beolvasása a klaszterezéshez
data = np.genfromtxt('kmeans_data2.txt')
In [10]:
plt.figure(figsize=(20, 10), dpi=300)
plt.scatter(data[:,0], data[:,1], color='red')
plt.show()