diff options
| -rw-r--r-- | alex/plots.py | 29 | ||||
| l--------- | data/test.csv | 1 | ||||
| l--------- | data/train.csv | 1 |
3 files changed, 31 insertions, 0 deletions
diff --git a/alex/plots.py b/alex/plots.py new file mode 100644 index 0000000..e405480 --- /dev/null +++ b/alex/plots.py @@ -0,0 +1,29 @@ +import matplotlib.pyplot as plt +import numpy +import cPickle +import scipy + +print "Loading data..." +with open("train_normal.pkl") as f: normal = cPickle.load(f) + +print "Extracting x and y" +xes = [c[0] for l in normal for c in l[-1]] +yes = [c[1] for l in normal for c in l[-1]] + +xrg = [-8.75, -8.55] +yrg = [41.05, 41.25] + +print "Doing 1d histogram" +#plt.clf(); plt.hist(xes, bins=1000, range=xrg); plt.savefig("xhist.pdf") +#plt.clf(); plt.hist(yes, bins=1000, range=yrg); plt.savefig("yhist.pdf") + +print "Doing 2d histogram" +#plt.clf(); plt.hist2d(xes, yes, bins=500, range=[xrg, yrg]); plt.savefig("xymap.pdf") + +hist, xx, yy = numpy.histogram2d(xes, yes, bins=2000, range=[xrg, yrg]) + +import ipdb; ipdb.set_trace() + +plt.clf(); plt.imshow(numpy.log(hist)); plt.savefig("xyhmap.pdf") + +scipy.misc.imsave("xymap.png", numpy.log(hist)) diff --git a/data/test.csv b/data/test.csv new file mode 120000 index 0000000..b797a91 --- /dev/null +++ b/data/test.csv @@ -0,0 +1 @@ +/data/lisatmp3/auvolat/taxikaggle/test.csv
\ No newline at end of file diff --git a/data/train.csv b/data/train.csv new file mode 120000 index 0000000..d394bdf --- /dev/null +++ b/data/train.csv @@ -0,0 +1 @@ +/data/lisatmp3/auvolat/taxikaggle/train.csv
\ No newline at end of file |