mnistsom.py

from som import SOM
import numpy as np
import math
from mnistgraphics import MNISTGraphics
import random

class MNISTSom(SOM):
    def __init__(self, train_x, train_y, test_x, test_y, nodes, sigma_0, learn_rate_0, total_iterations, learn_rate_timeconst, scaler, sigma_timeconst, plot_interval):
        SOM.__init__(self, train_x, nodes, sigma_0, learn_rate_0, sigma_timeconst, learn_rate_timeconst, total_iterations, plot_interval = plot_interval, graphics = MNISTGraphics())
        self.scaler = scaler
        self.train_y = train_y
        self.test_x = test_x
        self.test_y = test_y


    def train(self):

        while self.iteration <= self.total_iterations:
            for data in self.dataset:
                closest_node = self.closest_node(data)
                self.update_nodes(closest_node, data)

            if(self.iteration % self.plot_interval == 0):
                self.update_labels()
                self.graphics.draw_frame(self, self.iteration)
                self.test(type='train')
                print("Learn rate {}".format(self.learn_rate()))
                print("Sigma {}".format(self.sigma()))


            self.iteration += 1
        self.graphics.wait()




    def test(self, type = 'train'):
        if type == 'train':
            data_x = self.dataset
            data_y = self.train_y
        elif type == 'test':
            data_x = self.test_x
            data_y = self.test_y
        else:
            return

        correct = 0
        for data, label in zip(data_x, data_y):
            closest_node = self.closest_node(data)
            if closest_node.get_number() == label:
                correct += 1
        print("{} accuracy: {}".format(type,correct/len(data_x)))

    def update_labels(self):
        for node in self.nodes:
            node.reset_labels()

        for data, label in zip(self.dataset, self.train_y):
            closest_node = self.closest_node(data)
            closest_node.add_label(label)
        print("updating labels done")


    def report(self):
        print("Training done with results:\n")
        self.test(type='train')
        self.test(type='test')