def test_KNM(): dane = pobierz_dane() dane = przygotuj_dane(dane, [1,6], 3, [3,4,5]) ostatnie = pobierz_ostatnie_k_zakupow(1, 141L) ostatnie = przygotuj_dane(ostatnie, [1,6], 3, [3,4,5]) transakcja = ostatnie[0] k_sasiadow = 20 kol_decyz = 2 indeksy = [3,4,5,6] pol = KNN(dane, k_sasiadow, transakcja, kol_decyz, indeksy) print pol
def znajdz_polecane_dla_k_ostatnich(K, id_klienta): dane = pobierz_dane() dane = przygotuj_dane(dane, [1,6], 3, [3,4,5]) k_sasiadow = 20 kol_decyz = 2 # indeksy sa to zmienne kolumn brane pod uwagę, w KNN # productLine, productScale, productVendor, buyPrice indeksy = [3,4,5,6] ostatnie = pobierz_ostatnie_k_zakupow(10, id_klienta) ostatnie = przygotuj_dane(ostatnie, [1,6], 3, [3,4,5]) polecenia = Counter() for transakcja in ostatnie: pol = KNN(dane, k_sasiadow, transakcja, kol_decyz, indeksy) polecenia += pol return polecenia
def znajdz_polecane_dla_k_ostatnich(K, id_klienta): dane = pobierz_dane() dane = przygotuj_dane(dane, [1, 6], 3, [3, 4, 5]) k_sasiadow = 20 kol_decyz = 2 # indeksy sa to zmienne kolumn brane pod uwagę, w KNN # productLine, productScale, productVendor, buyPrice indeksy = [3, 4, 5, 6] ostatnie = pobierz_ostatnie_k_zakupow(10, id_klienta) ostatnie = przygotuj_dane(ostatnie, [1, 6], 3, [3, 4, 5]) polecenia = Counter() for transakcja in ostatnie: pol = KNN(dane, k_sasiadow, transakcja, kol_decyz, indeksy) polecenia += pol return polecenia
def test_K_ostatnich(): for id in [125]: ostatnie = pobierz_ostatnie_k_zakupow(8, id) print len(ostatnie) for o in ostatnie: print o