IA 29.08.17

#https://notepad.pw/uni7-ia-manha-knn-exercicio

from random import random, randint
import math

def wineprice(rating, age):
    peak_age = rating-50
    price = rating/2
    if (age > peak_age):
        price = price*(5-(age-peak_age))
    else:
        price = price*(5*((age+1)/peak_age))
    if price < 0:
        price = 0
    return price

def winedataset(size):
    raws = []
    for i in range(size):
        rating = random()*50+50
        age = random()*50
        price = wineprice(rating, age)
        price *= (random()*0.4+0.8)
        raws.append({'input':(rating,age),
                    'result':price})
    return raws

def euclidean(v1,v2):
    d = 0.0
    for i in range(len(v1)):
        d += (v1[i]-v2[i])**2
    return math.sqrt(d)

def getdistances(data, vec1):
    distancelist = []
    for i in range(len(data)):
        vec2 = data[i]['input']
        distancelist.append((euclidean(vec1, vec2),i))
    distancelist.sort()
    return distancelist

def knnestimate(data, vec1):
    # Get sorted distances
    dlist = getdistances(data,vec1)
    avg = 0.0
    avglist = []
    # Take the average of the top k results
    for k in range(1,16):
      for i in range(k):
        idx = dlist[i][1]
        avg += data[idx]['result']
      avg = avg / k
      avglist.append(avg)
    for j in avglist:
      if j == min(avglist):
        return avglist.index(j)+1

#Dataset de Treinamento
data_test = winedataset(70)
#Dataset de Teste
data_real = winedataset(30)
print(knnestimate(data_test,(95.0,3.0)))

print("### BANCO DE TESTE ###")
for i in data_test:
    print("Preco\t|\tClassificacao\t| Idade\t| Valor K")
    print('{:.2f}\t|\t{:.2f}\t        | {:.2f}\t {}'.format(i['result'],i['input'][0],i['input'][1], knnestimate(data_test,(i['input'][0], i['input'][1]))))

print("\n### BANCO REAL ###")
for i in data_real:
    print("Preco\t|\tClassificacao\t| Idade")
    print('{:.2f}\t|\t{:.2f}\t        | {:.2f}'.format(i['result'],i['input'][0],i['input'][1]))

#dist = euclidean(data[0]['input'], data[1]['input'])
#print(dist)