Skip to content

yyt17/Input-aware-Factorization-Machine

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 

Repository files navigation

Input-aware Factorization Machine (IFM)

This repository provides an implementation and datasets of IFM which is a novel factorization machine model. In addition, some additional experiments not included in the paper are also presented here, including AUC performance test on Avazu dataset, and so on.

Code introduction

The code is a Python implementation of IFM.

Data description

The Frappe dataset has been used for context-aware mobile app recommendation, which contains 96,202 records containing 957 users and 4,082 apps.

The MovieLens dataset has been used for personalized tag recommendation, which contains 668,953 tag applications of 17,045 users on 23,743 items with 49,657 distinct tags.

About

An Input-aware Factorization Machine for Sparse Prediction

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%