+ '_' + datetime.datetime.now().strftime("%Y-%m-%d_%H-%M")
if mode=="offline":
complete_name+="_test="+str(test_num)
bot = Bot_v1(complete_name)
try:
######################SHRINKED
dr = Datareader(mode=mode, test_num=test_num, train_format="50k", only_load=True)
ev = Evaluator(dr)
pids = dr.get_test_pids()
urm, dictns, dict2 = dr.get_urm_shrinked()
urm_evaluation = dr.get_evaluation_urm()[pids]
pids_converted = np.array([dictns[x] for x in pids], dtype=np.int32)
slim = MultiThreadDSLIM_RMSE(urm.T)
slim.fit(l1_penalty=l1, l2_penalty=l2, positive_only=True, beta=beta, topK=topk)
evaluate_shrinked(W_sparse= slim.W_sparse, urm_shrinked= urm, pids_shrinked= pids_converted)
sps.save_npz(complete_name+".npz",slim.W_sparse,)
except Exception as e:
bot.error("Exception "+str(e))
bot.end()
verbose=True,
train_format=train_format,
only_load=True)
if mode == 'offline':
if len(train_format > 0):
####### DATA INIZIALIZATION SHRINKED #################
dr = Datareader(mode=mode,
train_format=train_format,
only_load=True)
ev = Evaluator(dr)
pids = dr.get_test_pids()
urm, dict_n_to_s, dict2 = dr.get_urm_shrinked()
urm_evaluation = dr.get_evaluation_urm()
urm_evaluation = urm_evaluation[pids]
pids_converted = np.array([dict_n_to_s[x] for x in pids],
dtype=np.int32)
else:
####### DATA INIZIALIZATION FULL #################
dr = Datareader(mode=mode, only_load=True, verbose=False)
ev = Evaluator(dr)
pids = dr.get_test_pids()
urm = dr.get_urm()
urm_evaluation = dr.get_evaluation_urm()
urm_evaluation = urm_evaluation[pids]
urm_evaluation = None
pids_converted = pids