Skip to content

Code for AAAI21 paper "Scalable and Explainable 1-Bit Matrix Completion via Graph Signal Learning"

License

Notifications You must be signed in to change notification settings

cchao0116/SGMC-AAAI21

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SGMC: Spectral Graph Matrix Completion

Code for AAAI21 paper "Scalable and Explainable 1-Bit Matrix Completion via Graph Signal Learning".

Data Format

The implementation is desiged for top-N recommendations on implicit data, and thus it takes user-item pairs as input:

uid,sid
1,1

Installation

The program requires Python 3.7+ with NumPy, SciPy, Pandas and PySpark.

Note:

Train and Test

After specifying the location of files train.csv/test_tr.csv/test_te.csv in runme.sh, it is quite simple to train and evaluate the model by

bash runme.sh

Citation

If you find our code useful for your research, please consider cite.

@inproceedings{chen2021scalable,
  title={Scalable and Explainable 1-Bit Matrix Completion via Graph Signal Learning},
  author={Chen, Chao and Li, Dongsheng and Yan, Junchi and Huang, Hanchi and Yang, Xiaokang},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence (AAAI '21)},
  volume={35},
  number={8},
  pages={7011--7019},
  year={2021}
}

About

Code for AAAI21 paper "Scalable and Explainable 1-Bit Matrix Completion via Graph Signal Learning"

Topics

Resources

License

Stars

Watchers

Forks