ncf_single: Neural Collaborative Filtering with a single configuration

In tianwei-zhang/easyAI: Democratize AI

Description

Neural Collaborative Filtering with a single configuration

Usage

ncf_single(data, mf_output_dim, mlp_complexity, lr, epoch, top)

Arguments

data	Transaction dataset. Format must be order_id (could be dummy), user_id, and product_id
mf_output_dim	Number of latent vectors to represent users and items
mlp_complexity	Number of hidden layers in the MLP layer
top	Number of top products and customers to include. Can be NULL (e.g. will include everything)

Value

returns a list object with three values: model: keras model contructed. A keras_model object loss: a vector containing loss value in each epoch

tianwei-zhang/easyAI documentation built on May 14, 2019, 12:48 p.m.