Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
src
 
 
 
 
 
 
 
 

README.md

Nonsymmetric DPP Learning

PyTorch implementation of nonsymmetric determinantal point process (DPP) learning. For details, see our Learning Nonsymmetric Determinantal Point Processes paper (NeurIPS 2019 poster).

Installation

Install required Python packages:

$ pip install --user --requirement requirements.txt

Usage

Train and evaluate a nonsymmetric DPP model using the Amazon apparel baby registry dataset:

$ cd src
$ python main.py --dataset_name basket_ids --input_file data/1_100_100_100_apparel_regs.csv --num_sym_embedding_dims 30 --num_nonsym_embedding_dims 30

For a full list of command-line options:

$ python main.py --help

About

PyTorch implementation of nonsymmetric determinantal point process (DPP) learning.

Topics

Resources

License

Releases

No releases published

Packages

No packages published

Languages

You can’t perform that action at this time.