Mercurial > hg > dml-open-backendtools
comparison pyspark/test_timeside_vamp_spark.py @ 0:e34cf1b6fe09 tip
commit
author | Daniel Wolff |
---|---|
date | Sat, 20 Feb 2016 18:14:24 +0100 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:e34cf1b6fe09 |
---|---|
1 # Part of DML (Digital Music Laboratory) | |
2 # | |
3 # This program is free software; you can redistribute it and/or | |
4 # modify it under the terms of the GNU General Public License | |
5 # as published by the Free Software Foundation; either version 2 | |
6 # of the License, or (at your option) any later version. | |
7 # | |
8 # This program is distributed in the hope that it will be useful, | |
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
11 # GNU General Public License for more details. | |
12 # | |
13 # You should have received a copy of the GNU General Public | |
14 # License along with this library; if not, write to the Free Software | |
15 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
16 | |
17 #!/usr/local/spark-1.0.0-bin-hadoop2/bin/spark-submit | |
18 # -*- coding: utf-8 -*- | |
19 __author__="wolffd" | |
20 __date__ ="$11-Jul-2014 15:31:01$" | |
21 | |
22 from pyspark import SparkConf, SparkContext | |
23 # @todo: timeside has to be packed for multi-pc usage | |
24 from timeside_vamp import * | |
25 from os import walk | |
26 | |
27 # Running python applications through ./bin/pyspark is deprecated as of Spark 1.0. | |
28 # Use ./bin/spark-submit <python file> | |
29 | |
30 | |
31 def main(): | |
32 print "PySpark Telemeta and Vamp Test" | |
33 conf = (SparkConf() | |
34 .setMaster("local") | |
35 .setAppName("My app") | |
36 .set("spark.executor.memory", "1g")) | |
37 sc = SparkContext(conf = conf) | |
38 | |
39 # here come the wav file names | |
40 | |
41 mypath = '../../TimeSide/tests/samples/' | |
42 data = [] | |
43 for (dirpath, dirnames, filenames) in walk(mypath): | |
44 for file in filenames: | |
45 if file.endswith(".wav"): | |
46 data.append(os.path.join(dirpath, file)) | |
47 | |
48 # define distributed dataset | |
49 distData = sc.parallelize(data) | |
50 | |
51 # define map | |
52 m1 = distData.map(lambda x: transform(wav_file=x)) | |
53 | |
54 #process 2 | |
55 m1.take(2) | |
56 | |
57 if __name__ == "__main__": | |
58 main() | |
59 |