loadsvm.py

import numpy as np

# goran = 0 .....  georgi =  1 ..... gorjan  = 2 ..... ivan = 3
def normalize(x):
    x_norm = x
    num_col = x.shape[1]
    for i in range(0,num_col):
        x_norm[:, i] = (x_norm[:, i] - np.mean(x[:, i]))/np.std(x[:, i])
    return x_norm


def data():
    n = 13
    brojnesto = 0
    kolkuceka =0
    kolkuceka500 = 0
    kt = 0
    kolkulevo = kolkudesno = kolkugore = kolkudole = 0
    kolkutrening = 0
    rastprva = 0
    vkupnovreme = 0
    brzina = brojac = 0
    agolminus = agolplus= 0
    brgore = brdole = brdesno = brlevo = 1
    trX = np.zeros([1151, n])
    trY = np.zeros([1151],dtype=np.int8)
    teX = np.zeros([192, n])
    teY = np.zeros([192],dtype=np.int8)
    lines = [line.rstrip('\n') for line in open('hw2_mouse_data.txt')]
    for i in range(0, len(lines)-1):
        linei = lines[i]
        linei = linei[1:len(linei) - 1]
        splitovano = linei.split(', ')
        vlezi = np.asarray(splitovano)
        if vlezi[0] == 'Goran' or vlezi[0] == 'Georgi' or vlezi[0] == 'Gorjan' or vlezi[0] == 'Ivan':
            if vlezi[0] == 'Goran':
                izlez = 0
            if vlezi[0] == 'Georgi':
                izlez = 1
            if vlezi[0] == 'Gorjan':
                izlez = 2
            if vlezi[0] == 'Ivan':
                izlez = 3
            prva = lines[i+1]
            prva = prva[1:len(prva) - 1]
            splitprva = prva.split(', ')
            prva = np.asarray(splitprva, dtype=float)
            i = i +1
            vlez = lines[i]
            vlez = vlez[1:len(vlez) - 1]
            splitovano = vlez.split(', ')
            vlez = np.asarray(splitovano)
            while len(vlez)> 1:
                line1 = lines[i]
                line1 = line1[1:len(line1) -1]
                splitovano = line1.split(', ')
                line1 = np.asarray(splitovano, dtype=float)
                i = i+1
                if i > (len(lines) - 1):
                    break
                pom = lines[i]
                pom = pom[1:len(pom)-1]
                splitovano = pom.split(', ')
                pom = np.asarray(splitovano)
                if len(pom) >1:
                    line2 = lines[i]
                    line2 = line2[1:len(line2) - 1]
                    splitovano = line2.split(', ')
                    line2 = np.asarray(splitovano, dtype=float)
                else:
                    break
                    vlez = pom

                x_razlika = line1[0] - line2[0]
                y_razlika = line1[1] - line2[1]
                rastprva += np.sqrt((prva[0] - line1[0])**2 + (prva[1] - line1[1])**2)
                if x_razlika:
                    agol= np.arctan(np.abs(y_razlika)/np.abs(x_razlika))
                else: agol = 1.57079623
                if (x_razlika >= 0 and y_razlika >= 0) or (x_razlika <= 0 and y_razlika <= 0):
                    agolplus += agol

                elif (x_razlika >=0 and y_razlika <=0) or (x_razlika<=0 and y_razlika>= 0):
                    agolminus += agol

                vreme = line2[2]
                if x_razlika > 0:
                    kolkulevo += np.abs(x_razlika)
                    brlevo += 1
                if x_razlika < 0:
                    kolkudesno += np.abs(x_razlika)
                    brdesno += 1
                if y_razlika > 0:
                    kolkudole += np.abs(y_razlika)
                    brdole += 1
                if y_razlika < 0:
                    kolkugore += np.abs(y_razlika)
                    brgore += 1
                brzina += np.sqrt(x_razlika ** 2 + y_razlika ** 2) / float(vreme)
                if vreme > 200:
                    kolkuceka500 += 1
                if vreme < 80:
                    vkupnovreme += vreme
                else:
                    vkupnovreme += 1
                    kolkuceka +=1
                kolkutrening += 1

            brzina = float(brzina) / float(kolkutrening)
            rastprva = float(rastprva) / float(kolkutrening)
            if kolkulevo:
                kolkulevo = float(kolkulevo) / float(brlevo)
            if kolkudesno:
                kolkudesno = float(kolkudesno) / float(brdesno)
            if kolkugore:
                kolkugore = float(kolkugore) / float(brgore)
            if kolkudole:
                kolkudole = float(kolkudole) / float(brdole)
            prosecnovreme = float(vkupnovreme) / float(kolkutrening)

            zabrzuvanje = float(brzina) / float(prosecnovreme)
            jerk = float(zabrzuvanje) / float (prosecnovreme)
            agolplus = float(agolplus)/ float(kolkutrening)
            agolminus = float(agolminus) / float(kolkutrening)

            niza = np.asarray(
                [kolkugore, kolkudole, kolkulevo, kolkudesno, brzina, np.sqrt(prosecnovreme), kolkuceka, kolkuceka500, agolplus, agolminus,
                 np.sqrt(kolkugore), zabrzuvanje, np.sqrt(jerk)])
            if brojnesto % 6 != 0:
                trX[brojac] = niza
                trY[brojac] = izlez
                brojac +=1
                brojnesto += 1
            else:
                teX[kt] = niza
                teY[kt] = izlez
                trX[brojac] = niza
                trY[brojac] = izlez
                brojac += 1
                kt += 1
                brojnesto +=1

            brzina = kolkutrening= vkupnovreme = agol =agolplus = agolminus =0
            kolkulevo = kolkudesno = kolkugore = kolkudole = rastprva = kolkuceka = kolkuceka500 = 0
            brdole = brgore = brdesno = brlevo = 1
    trX = normalize(trX)
    teX = normalize(teX)

    return trX, teX, trY, teY, n

data()