Merge pull request #10 from RUCAIBox/master

updates fork
RUCAIBox · Jul 19, 2020 · 97a1f4c · 97a1f4c
2 parents 6af995d + 1caf594
commit 97a1f4c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/model/context_aware_recommender/nfm.py b/model/context_aware_recommender/nfm.py
@@ -40,7 +40,7 @@ def __init__(self, config, dataset):
         self.first_order_linear = FMFirstOrderLinear(self.field_dims, self.offsets)
         self.embedding = FMEmbedding(self.field_dims, self.offsets, self.embedding_size)
         self.fm = BaseFactorizationMachine(reduce_sum=False)
-        self.mlp_layers = MLPLayers(self.layers, self.dropout, activation='sigmoid')
+        self.mlp_layers = MLPLayers(size_list, self.dropout, activation='sigmoid')
         self.predict_layer = nn.Linear(self.mlp_hidden_size[-1], 1, bias=False)
         self.sigmoid = nn.Sigmoid()
         self.loss = nn.BCELoss()
@@ -69,7 +69,9 @@ def forward(self, interaction):
             # todo: check (batch) or (batch, 1)
             x.append(interaction[field].unsqueeze(1))
         x = torch.cat(x, dim=1)
-        y = self.predict_layer(self.mlp_layers(self.embedding(x)))+self.first_order_linear(x)
+        emb_x = self.fm(self.embedding(x))
+
+        y = self.predict_layer(self.mlp_layers(emb_x))+self.first_order_linear(x)
         y = self.sigmoid(y)
         return y.squeeze()
 

diff --git a/properties/model/NFM.config b/properties/model/NFM.config
@@ -0,0 +1,5 @@
+[model]
+
+embedding_size=64
+mlp_hidden_size=[64, 64, 64]
+dropout=0.0