#all_models['S_SLIM_BPR'] = new_best_models.SSLIM_BPR.get_model(sps.vstack([URM_train, ICM_all.T]))
all_models['S_PURE_SVD'] = new_best_models.PureSVDSideInfo.get_model(URM_train, ICM_all)
all_models['S_IALS'] = new_best_models.IALSSideInfo.get_model(URM_train, ICM_all)
all_models['USER_CBF_CF'] = new_best_models.UserCBF_CF_Warm.get_model(URM_train, UCM_all)
all_models['USER_CF'] = new_best_models.UserCF.get_model(URM_train)
return all_models
if __name__ == '__main__':
# Data loading
root_data_path = "../../data/"
data_reader = RecSys2019Reader(root_data_path)
data_reader = New_DataSplitter_leave_k_out(data_reader, k_out_value=3, use_validation_set=False,
force_new_split=True, seed=get_split_seed())
data_reader.load_data()
URM_train, URM_test = data_reader.get_holdout_split()
# Build ICMs
ICM_all, _ = get_ICM_train_new(data_reader)
# Build UCMs
UCM_all = get_UCM_train(data_reader)
model = HybridRankBasedRecommender(URM_train)
all_models = _get_all_models(URM_train=URM_train, ICM_all=ICM_all, UCM_all=UCM_all)
for model_name, model_object in all_models.items():
model.add_fitted_model(model_name, model_object)
print("The models added in the hybrid are: {}".format(list(all_models.keys())))
def main():
args = get_arguments()
# Data loading
data_reader = RecSys2019Reader(args.reader_path)
data_reader = New_DataSplitter_leave_k_out(data_reader,
k_out_value=3,
use_validation_set=False,
force_new_split=True,
seed=args.seed)
data_reader.load_data()
URM_train, URM_test = data_reader.get_holdout_split()
ICM_numerical, _ = get_ICM_numerical(data_reader.dataReader_object)
similarity_type_list = None
# Setting evaluator
exclude_cold_users = args.exclude_users
h = int(args.focus_on_high)
fol = int(args.focus_on_low)
if h != 0:
print("Excluding users with less than {} interactions".format(h))
ignore_users_mask = np.ediff1d(URM_train.tocsr().indptr) < h
ignore_users = np.arange(URM_train.shape[0])[ignore_users_mask]
elif fol != 0:
print("Excluding users with more than {} interactions".format(fol))
warm_users_mask = np.ediff1d(URM_train.tocsr().indptr) > fol
ignore_users = np.arange(URM_train.shape[0])[warm_users_mask]
if exclude_cold_users:
cold_user_mask = np.ediff1d(URM_train.tocsr().indptr) == 0
cold_users = np.arange(URM_train.shape[0])[cold_user_mask]
ignore_users = np.unique(np.concatenate(
(cold_users, ignore_users)))
elif exclude_cold_users:
print("Excluding cold users...")
cold_user_mask = np.ediff1d(URM_train.tocsr().indptr) == 0
ignore_users = np.arange(URM_train.shape[0])[cold_user_mask]
else:
ignore_users = None
cutoff_list = [10]
evaluator = EvaluatorHoldout(URM_test,
cutoff_list=cutoff_list,
ignore_users=ignore_users)
# HP tuning
print("Start tuning...")
version_path = "../../report/hp_tuning/{}/".format(args.recommender_name)
now = datetime.now().strftime('%b%d_%H-%M-%S')
now = now + "_k_out_value_3/"
version_path = version_path + "/" + now
run_parameter_search_VNN(
URM_train=URM_train,
recommender_class=RECOMMENDER_CLASS_DICT[args.recommender_name],
evaluator_validation=evaluator,
metric_to_optimize="MAP",
output_folder_path=version_path,
similarity_type_list=similarity_type_list,
parallelizeKNN=True,
n_cases=int(args.n_cases),
n_random_starts=int(args.n_random_starts))
print("...tuning ended")