Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
xuxinzhang committed Jun 3, 2020
1 parent e28fe47 commit 85d0a5f
Showing 1 changed file with 15 additions and 1 deletion.
16 changes: 15 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,19 @@
# CFGAN
Implementation of CFGAN. The implementation based on Pytorch is done, and the max precision on ML100k is 0.49 in 1000 train epochs.
Implementation of CFGAN. The implementation based on Pytorch is done, and the max precision on ML100k is 0.49 in 1000 train epochs, when use the paramaters in the paper.

cfgan.py : the model of CFGAN

data.py : load data

main.py : train and show result


Run the demo :

Python main.py

You can modify paramaters in main.py, for example, modify the alpha to 0.5(in fact, result turn better when you do that)


Next, we consider implementing another version based on Tensorflow.

Expand Down

0 comments on commit 85d0a5f

Please sign in to comment.