Skip to content

USC-Melady/KDD19-CoSTCo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

KDD19-CoSTCo

Demo run for KDD2019 paper "CoSTCo: A Neural Tensor Completion Model for Sparse Tensors".

Requirements

We run experiments with following package versions with Python2.

tensorflow==1.14.0
keras==2.2.4

Run the demo

Run in the Google's colab (no need to install any packages)

  1. Open this colab link.
  2. Choose property runtime: Click "Runtime" -> "Change Runtime Type" -> "python=2, hardware accelerator=GPU"
  3. Run all: Click "Runtime" -> "Run all"

Run in a local machine

  1. git clone this repo
  2. (optional) set up virtualenv
  3. install required packages and jupyter-notebook
  4. open jupyter-notebook server and run the notebook "costco_demo.ipynb"