-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathscheduler.py
108 lines (78 loc) · 2.94 KB
/
scheduler.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon Aug 5 00:59:11 2019
@author: nicolas
"""
from ecommerce_postgresql import EcommercePostgres
#-----
import turicreate as tc
import schedule
import time
def do_n():
ecommerce_postgres = EcommercePostgres()
ratings = ecommerce_postgres.getDataframe()
'''
-----------------------------------------------------------------------------
Preprocesamiento de los datos utilizando tc
-----------------------------------------------------------------------------
'''
'''
Convirtiendo a SFrame para la entrada del modelo
'''
data = tc.SFrame(ratings)
'''
-----------------------------------------------------------------------------
Separando datos para evaluación del modelo
-----------------------------------------------------------------------------
'''
training_data, validation_data = tc.recommender.util.random_split_by_user(data, 'user_id', 'product_id')
'''
-----------------------------------------------------------------------------
Creando Modelo:
-----------------------------------------------------------------------------
'''
# Turi Create proporciona un método turicreate.recommender.create
# que elegirá automáticamente un modelo apropiado para su conjunto de datos.
model = tc.recommender.create(training_data, 'user_id', 'product_id',target='rating')
'''
----> Making recommendations for ALL USERS:
'''
print("##############################################")
print("ALL USERS:")
results = model.recommend()
print(results)
'''
----> Making recommendations for NEW USER:
'''
# Esto se conoce como el problema de "arranque en frío".
# La función recomendar () funciona a la perfección con los nuevos usuarios.
# Si el modelo nunca ha visto al usuario, entonces se recomienda elementos populares:
print("##############################################")
print("NEW USER:")
results1 = model.recommend(['-1'])
print(results1);
'''
----> Making recommendations for SPECIFIC USERS
'''
#QUERY BD:
#SELECT user_id,product_id,rating FROM reviews WHERE user_id=2;
print("##############################################")
users=['2']
print("USER: "+''.join(users))
recommendations = model.recommend(users)
print(recommendations);
'''
-----------------------------------------------------------------------------
SAVING model:
-----------------------------------------------------------------------------
'''
model.save("my_model.model")
'''
----> Obtener resumen del modelo recado con turicreate.recommender.create
'''
model.summary()
schedule.every(5).minutes.do(do_n)
while 1:
schedule.run_pending()
time.sleep(1)