Python CSVDataExtractor.extract_pl Examples

Example #1

File: swsort.py Project: dymx101/amiwork

from dataio.csvdataextractor import CSVDataExtractor
from recommendationsorters.simpleweightedsorter import SimpleWeightedSorter

if len(sys.argv) < 3:
    print "2 cmd parameters required. Usage:\npy dataextract.py <playlist_file> <recommendations_file> [include_artist (Possible values=yes)]"
    exit(1)

plfile = open(sys.argv[1])
recofile = open(sys.argv[2])

if not (plfile and recofile):
    print "One or more of the files were not found"
    exit(1)

extractor = CSVDataExtractor()
pl = extractor.extract_pl(plfile)
reco = extractor.extract_reco(recofile)

relevant_fields = ['genre', 'mood']
if len(sys.argv) >= 4 and sys.argv[3] == "yes":
    relevant_fields.append('artist')

swsorter = SimpleWeightedSorter()
swsorter.set_user_playlist(pl)
swsorter.set_recommendations(reco)

swsorter.set_relevant_fields(relevant_fields)

#print pl
#print reco

Example #2

File: ftsort.py Project: krishnangovindraj/amiwork

from recommendationsorters.ftsorter import FTSorter


if len(sys.argv) < 3:
	print "2 cmd parameters required. Usage:\npy dataextract.py <playlist_file> <recommendations_file> [include_artist (Possible values=yes)]"
	exit(1)

plfile = open(sys.argv[1])
recofile = open(sys.argv[2])

if not (plfile and recofile):
	print "One or more of the files were not found"
	exit(1)

extractor = CSVDataExtractor()
user_playlist = extractor.extract_pl(plfile)
recommendations = extractor.extract_reco(recofile)



relevant_fields = ['genre','mood']
if len(sys.argv)>=4 and sys.argv[3] == "yes":
		relevant_fields.append('artist')

ftsorter = FTSorter()
ftsorter.set_user_playlist(user_playlist)
ftsorter.set_recommendations(recommendations)
ftsorter.set_relevant_fields( relevant_fields )#,"artist"] )
ftsorter.build_user_ftree()

ftree= ftsorter.user_ftree

Example #3

from dataio.csvdataextractor import CSVDataExtractor
from recommendationsorters.ftsorter import FTSorter

if len(sys.argv) < 3:
    print "2 cmd parameters required. Usage:\npy dataextract.py <playlist_file> <recommendations_file> [include_artist (Possible values=yes)]"
    exit(1)

plfile = open(sys.argv[1])
recofile = open(sys.argv[2])

if not (plfile and recofile):
    print "One or more of the files were not found"
    exit(1)

extractor = CSVDataExtractor()
user_playlist = extractor.extract_pl(plfile)
recommendations = extractor.extract_reco(recofile)

relevant_fields = ['genre', 'mood']
if len(sys.argv) >= 4 and sys.argv[3] == "yes":
    relevant_fields.append('artist')

ftsorter = FTSorter()
ftsorter.set_user_playlist(user_playlist)
ftsorter.set_recommendations(recommendations)
ftsorter.set_relevant_fields(relevant_fields)  #,"artist"] )
ftsorter.build_user_ftree()

ftree = ftsorter.user_ftree
#ftree.print_tree()

Example #4

File: dataextract.py Project: krishnangovindraj/amiwork

import sys
from models.song import Song
from dataio.csvdataextractor import CSVDataExtractor

if len(sys.argv) < 3:
	print "2 cmd parameters required. Usage:\npy dataextract.py <playlist_file> <recommendations_file>"
	exit(1)

plfile = open(sys.argv[1])
recofile = open(sys.argv[2])

if not (plfile and recofile):
	print "One or more of the files were not found"
	exit(1)

extractor = CSVDataExtractor()
pl = extractor.extract_pl(plfile)
reco = extractor.extract_reco(recofile)

print pl
print reco