Skip to content

Commit

Permalink
Update block comment
Browse files Browse the repository at this point in the history
  • Loading branch information
darrylong committed Oct 12, 2023
1 parent b77a268 commit 0591847
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions cornac/models/ngcf/recom_ngcf.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 0591847

Please sign in to comment.