arleilps · dsacc · Aug 28, 2017 · Aug 28, 2017 · Aug 28, 2017 · Aug 28, 2017
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,10 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# Uncompressed data
+data/
+
+# Jupyter Notebook
+.ipynb_checkpoints
diff --git a/.travis.yml b/.travis.yml
@@ -0,0 +1,18 @@
+# After changing this file, check it on:
+# http://lint.travis-ci.org
+
+language: python
+
+sudo: false
+
+python:
+  - "3.4"
+
+install:
+    - pip install -U pip
+    - pip install -U -r requirements.txt
+
+script:
+    - tar xvzf data.tar.gz
+    - export PYTHONHASHSEED=0
+    - python -m doctest lib/experiments.py
diff --git a/README.md b/README.md
@@ -2,7 +2,9 @@
 
 Implementation of graph wavelets via sparse cuts with some baselines, datasets and evaluation.
 
-Evaluation is performed using python notebooks.
+Evaluation is performed using IPython Notebook.
+
+After code review some results may differ from those presented in the paper.
 
 Scalability and approximation experiments:
 -----------------------
@@ -12,10 +14,30 @@ Compression experiments:
 -----------------------
 https://nbviewer.jupyter.org/github/arleilps/sparse-wavelets/blob/master/compression-experiments.ipynb
 
+Testing:
+------
+At the moment there is only one doctest in lib/experiments.py. To run the test
+you should use python version 3.4 or 3.5, NetworkX 1.11 and set PYTHONHASHSEED=0.
+This conditions are used to constrain the behaviour of the NetworkX function
+fiedler_vector(). Once ready, enter the following command:
+```
+python -m doctest lib/experiments.py -v
+```
+
+List of supported Python versions:
+------------------
+<ul>
+<li>2.7</li>
+<li>3.4</li>
+<li>3.5</li>
+<li>3.6</li>
+</ul>
+
+<br />
+
 For more details, see the paper:  
 [Graph Wavelets via Sparse Cuts ](http://arxiv.org/abs/1602.03320 "")  
 Arlei Silva, Xuan-Hong Dang, Prithwish Basu, Ambuj K Singh, Ananthram Swami  
-ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD), 2016 (to appear). 
+ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD), 2016.
 
 Arlei Silva (arlei@cs.ucsb.edu)
-
diff --git a/compression-experiments.ipynb b/compression-experiments.ipynb
diff --git a/figs/compression_blog.png b/figs/compression_blog.png
diff --git a/figs/compression_blog2.png b/figs/compression_blog2.png
diff --git a/figs/compression_human.png b/figs/compression_human.png
diff --git a/figs/compression_human2.png b/figs/compression_human2.png
diff --git a/figs/compression_small_traffic.png b/figs/compression_small_traffic.png
diff --git a/figs/compression_traffic.png b/figs/compression_traffic.png
diff --git a/figs/compression_wiki.png b/figs/compression_wiki.png
diff --git a/figs/compression_wiki2.png b/figs/compression_wiki2.png
diff --git a/lib/__init__.py b/lib/__init__.py
diff --git a/lib/datasets.py b/lib/datasets.py
@@ -1,35 +1,52 @@
+r"""
 
-#small traffic
-#Speeds from traffic sensors
-#vertices: 100
+This module provides path to the dataset in data.
+Each one is in a directory with the following structure:
+
+    datasetname
+      |- datasetname.data
+      |- datasetname.graph
+
+datasetname.data has info about the graph signal.
+Row format : "vertex_id, vertex_value"
+
+datasetname.graph has info about edges.
+Row format: "vertex_A, vertex_B[, edge_weight]"
+
+"""
+
+
+# Small traffic (weighted)
+# Speeds from traffic sensors
+# Vertices: 100
 
 small_traffic = {}
 small_traffic["path"] = "data/small_traffic/"
 
-#Large traffic
-#Speeds from traffic sensors
-#vertices: 1923
+# Large traffic (weigthed)
+# Speeds from traffic sensors
+# Vertices: 1923
 
 traffic = {}
 traffic["path"] = "data/traffic/"
 
-#Human
-#Gene expression data
-#vertices: 3628
+# Human (unweighted)
+# Gene expression data
+# Vertices: 3628
 
 human = {}
 human["path"] = "data/human/"
 
-#Wikipedia data
-#Number of views of wikipedia pages
-#Vertices: 4871
+# Wikipedia data (unweighted)
+# Number of views of wikipedia pages
+# Vertices: 4871
 
 wiki = {}
 wiki["path"] = "data/wiki/"
 
-#Political blogs
-#Link network of congressman's blogs with democrat/republican (0/1) as signal
-#vertices: 1490
+# Political blogs (unweighted)
+# Link network of congressman's blogs with democrat/republican (0/1) as signal
+# Vertices: 1490
 
 polblogs = {}
 polblogs["path"] = "data/polblogs/"