blob: 6333ea0f8925980b75e079a05d58bd8da836edb6 [file] [log] [blame]
itemrank quickstart import data
import predictionio
import random
import sys
def import_itemrank(access_key):
client = predictionio.EventClient(access_key)
print client.get_status()
# generate 10 users, with user ids 1,2,....,10
user_ids = [str(i) for i in range(1, 11)]
for user_id in user_ids:
print "Set user", user_id
# generate 50 items, with item ids 1,2,....,50
# assign type id 1 to all of them
item_ids = [str(i) for i in range(1, 51)]
for item_id in item_ids:
print "Set item", item_id
client.set_item(item_id, {
"itypes" : ['1']
# each user randomly views 10 items
for user_id in user_ids:
for viewed_item in random.sample(item_ids, 10):
print "User", user_id ,"views item", viewed_item
client.record_user_action_on_item("view", user_id, viewed_item)
if __name__ == '__main__':
if len(sys.argv) < 2:
sys.exit("Usage: python -m examples.itemrank_quick_start <access_key>")