TensorRec是Python中的TensorFlow推荐算法和框架

TensorRec是Python中的TensorFlow推荐算法和框架

TensorRec

A TensorFlow recommendation algorithm and framework in Python.

NOTE: TensorRec is not under active development

TensorRec will not be receiving any more planned updates. Please feel free to open pull requests -- I am happy to review them.

Thank you for your contributions, support, and usage of TensorRec!

-James Kirk, @jfkirk

For similar tools, check out:

TensorFlow Ranking

Spotlight

LightFM

What is TensorRec?

TensorRec is a Python recommendation system that allows you to quickly develop recommendation algorithms and customize them using TensorFlow.

TensorRec lets you to customize your recommendation system's representation/embedding functions and loss functions while TensorRec handles the data manipulation, scoring, and ranking to generate recommendations.

A TensorRec system consumes three pieces of data: user_features, item_features, and interactions. It uses this data to learn to make and rank recommendations.

For an overview of TensorRec and its usage, please see the wiki.

For more information, and for an outline of this project, please read this blog post.

For an introduction to building recommender systems, please see these slides.

Example: Basic usage

import numpy as npimport tensorrec# Build the model with default parametersmodel = tensorrec.TensorRec()# Generate some dummy datainteractions, user_features, item_features = tensorrec.util.generate_dummy_data( num_users=100, num_items=150, interaction_density=.05)# Fit the model for 5 epochsmodel.fit(interactions, user_features, item_features, epochs=5, verbose=True)# Predict scores and ranks for all users and all itemspredictions = model.predict(user_features=user_features, item_features=item_features)predicted_ranks = model.predict_rank(user_features=user_features, item_features=item_features)# Calculate and print the recall at 10r_at_k = tensorrec.eval.recall_at_k(predicted_ranks, interactions, k=10)print(np.mean(r_at_k))

Quick Start

TensorRec can be installed via pip: pip install tensorrec

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。