From 05918476970a628e696fe153ea7ca68b6985bd40 Mon Sep 17 00:00:00 2001 From: Darryl Ong Date: Thu, 12 Oct 2023 12:21:32 +0800 Subject: [PATCH] Update block comment --- cornac/models/ngcf/recom_ngcf.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/cornac/models/ngcf/recom_ngcf.py b/cornac/models/ngcf/recom_ngcf.py index 22ead5217..7284b8af7 100644 --- a/cornac/models/ngcf/recom_ngcf.py +++ b/cornac/models/ngcf/recom_ngcf.py @@ -21,24 +21,25 @@ class NGCF(Recommender): """ - NGCF + Neural Graph Collaborative Filtering Parameters ---------- name: string, default: 'NGCF' The name of the recommender model. - num_epochs: int, default: 1000 - Maximum number of iterations or the number of epochs. - emb_size: int, default: 64 Size of the node embeddings. layer_sizes: list, default: [64, 64, 64] Size of the output of convolution layers. - dropout_rates: list, defaukt: [0.1, 0.1, 0.1] - Dropout out rate for each of the convolution layers. + dropout_rates: list, default: [0.1, 0.1, 0.1] + Dropout rate for each of the convolution layers. + - Number of values should be the same as 'layer_sizes' + + num_epochs: int, default: 1000 + Maximum number of iterations or the number of epochs. learning_rate: float, default: 0.001 The learning rate that determines the step size at each iteration