From 530327a45f96a828459b5cf77d98321deb8e1d95 Mon Sep 17 00:00:00 2001 From: Xiang Wang Date: Fri, 19 Jun 2020 11:27:45 +0800 Subject: [PATCH] add 'raise NotImplementedError' in alg_type Raise NotImplementedError when alg_type is not set as one of ['bi', 'kgat', 'gcn', 'graphsage'], in order to solve some errors like 'no ua_embeddings'. --- Model/KGAT.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/Model/KGAT.py b/Model/KGAT.py index 17e0534..aaa8fd6 100644 --- a/Model/KGAT.py +++ b/Model/KGAT.py @@ -158,7 +158,7 @@ def _build_weights(self): return all_weights def _build_model_phase_I(self): - if self.alg_type in ['bi']: + if self.alg_type in ['bi', 'kgat']: self.ua_embeddings, self.ea_embeddings = self._create_bi_interaction_embed() elif self.alg_type in ['gcn']: @@ -166,6 +166,9 @@ def _build_model_phase_I(self): elif self.alg_type in ['graphsage']: self.ua_embeddings, self.ea_embeddings = self._create_graphsage_embed() + else: + print('please check the the alg_type argument, which should be bi, kgat, gcn, or graphsage.') + raise NotImplementedError self.u_e = tf.nn.embedding_lookup(self.ua_embeddings, self.users) self.pos_i_e = tf.nn.embedding_lookup(self.ea_embeddings, self.pos_items)