예제 #1
0
파일: NCF.py 프로젝트: youngzw/ReChorus 
 def parse_model_args(parser):
     parser.add_argument('--layers',
                         type=str,
                         default='[64]',
                         help="Size of each layer.")
     return BPR.parse_model_args(parser)
예제 #2
0
 def parse_model_args(parser):
     parser.add_argument('--time_bin',
                         type=int,
                         default=100,
                         help='Number of time bins.')
     return BPR.parse_model_args(parser)