• guptapriya's avatar
    Change batch size and epochs for NCF benchmarks · e8f97a1d
    guptapriya authored
    Current batch size 160000 does not converge to the desired HR. So we decrease to 99k which is known to converge. Tested locally and got to 63.5 at epoch 7. Also decreasing number of epochs as I don't see any improvement after epoch 7-8.
    e8f97a1d
ncf_keras_benchmark.py 5.44 KB