Mercurial > pylearn

annotate pylearn/datasets/test_tzanetakis.py @ 1479:1b69d435f09f

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
fix error string.
author Frederic Bastien <nouiz@nouiz.org>
date Wed, 25 May 2011 09:26:47 -0400
parents b39a9f02ef01
children
rev   line source
671
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
1 import theano
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
2
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
3 from pylearn.io import wavread
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
4 from pylearn.datasets import tzanetakis
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
5
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
6 def test_tzanetakis():
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
7 idx = theano.tensor.lscalar()
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
8
733
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
9 tz_ex = tzanetakis.TzanetakisExample()
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
10 path, label = tz_ex(idx)
671
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
11 print path, label
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
12
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
13 f = theano.function([idx], [path, label])
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
14
733
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
15 return
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
16
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
17 ### The code below does not run, because of
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
18 # TypeError: object of type 'TzanetakisExample' has no len()
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
19 #
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
20 # Someone should fix it if they want to finish the test! For now, it is
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
21 # just disabled (since the function returns above).
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
22
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
23 for i in xrange(len(tz_ex)):
671
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
24 print i, f(i)
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
25
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
26 wav,sr = wavread.wav_read_int16(path)
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
27
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
28 f = theano.function([idx], wav)
733
b39a9f02ef01 Test test_tzanetakis now passes
Olivier Delalleau <delallea@iro>
parents: 671
diff changeset
29 for i in xrange(len(tz_ex)):
671
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
30 print i, f(i).shape
9e62fd6b6677 adding wavread and tzanetakis dataset
James Bergstra <bergstrj@iro.umontreal.ca>
parents:
diff changeset
31