Ejemplo n.º 1
0
 def test_sasrecf(self):
     config_dict = {
         'model': 'SASRecF',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 2
0
 def test_autoint(self):
     config_dict = {
         'model': 'AutoInt',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 3
0
 def test_stamp(self):
     config_dict = {
         'model': 'STAMP',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 4
0
 def test_itemknn(self):
     config_dict = {
         'model': 'ItemKNN',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 5
0
 def test_widedeep(self):
     config_dict = {
         'model': 'WideDeep',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 6
0
 def test_xdeepfm(self):
     config_dict = {
         'model': 'xDeepFM',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 7
0
 def test_dssm(self):
     config_dict = {
         'model': 'DSSM',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 8
0
 def test_spectralcf(self):
     config_dict = {
         'model': 'SpectralCF',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 9
0
 def test_dcn(self):
     config_dict = {
         'model': 'DCN',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 10
0
 def test_fism(self):
     config_dict = {
         'model': 'FISM',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 11
0
 def test_nais(self):
     config_dict = {
         'model': 'NAIS',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 12
0
 def test_convncf(self):
     config_dict = {
         'model': 'ConvNCF',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 13
0
 def test_neumf(self):
     config_dict = {
         'model': 'NeuMF',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 14
0
 def test_kgat(self):
     config_dict = {
         'model': 'KGAT',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 15
0
 def test_sasrec(self):
     config_dict = {
         'model': 'SASRec',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 16
0
 def test_bert4rec(self):
     config_dict = {
         'model': 'BERT4Rec',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 17
0
 def test_transrec(self):
     config_dict = {
         'model': 'TransRec',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=config_file_list,
                        saved=False)
Ejemplo n.º 18
0
 def test_lightgcn(self):
     config_dict = {
         'model': 'LightGCN',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 19
0
 def test_gcmc(self):
     config_dict = {
         'model': 'GCMC',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 20
0
 def test_ngcf(self):
     config_dict = {
         'model': 'NGCF',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 21
0
 def test_kgnnls(self):
     config_dict = {
         'model': 'KGNNLS',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 22
0
 def test_ripplenet(self):
     config_dict = {
         'model': 'RippleNet',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 23
0
 def test_cfkg(self):
     config_dict = {
         'model': 'CFKG',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 24
0
 def test_fdsa(self):
     config_dict = {
         'model': 'FDSA',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 25
0
 def test_gru4recf(self):
     config_dict = {
         'model': 'GRU4RecF',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 26
0
 def test_nextitnet(self):
     config_dict = {
         'model': 'NextItNet',
         'reproducibility': False,
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 27
0
 def test_narm(self):
     config_dict = {
         'model': 'NARM',
         'loss_type': 'BPR',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 28
0
 def test_mkr(self):
     config_dict = {
         'model': 'MKR',
         'use_inner_product': False,
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 29
0
 def test_dmf(self):
     config_dict = {
         'model': 'DMF',
         'inter_matrix_type': 'rating',
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)
Ejemplo n.º 30
0
 def test_gcsan(self):
     config_dict = {
         'model': 'GCSAN',
         'MAX_ITEM_LIST_LENGTH': 3,
     }
     objective_function(config_dict=config_dict,
                        config_file_list=['test_model.yaml'],
                        saved=False)