コード例 #1
0
ファイル: ML_prep.py プロジェクト: pkundurthy/flopbuster
def retriveResponse(year=DEFAULT_YearDivide,is_training=True):
    """ Given the year dividing training and test sets, return the list of 
        all movies and the SuccessMetric """

    if is_training:
        comp_symbol = '<'
    else:
        comp_symbol = '>'

    statement1 = 'select title,released,ifnull(totalgross,0),'+\
                 'ifnull(budget,0)'+\
                 ' from boxoffice where boxoffice.released'+\
                 ' %s \"%s\" order by released asc;' % (comp_symbol,year)
    
    results = mysqlfuncs.db_execute_and_fetch(statement1)
    title = mysqlfuncs.results_to_list(results,index=0)
    released = mysqlfuncs.results_to_list(results,index=1)
    releaseYears = [long(misc.getYear(x.strftime("%Y-%m-%d"))) for x in released]

    totalGross = num.array(mysqlfuncs.results_to_list(results,index=2),\
                            dtype=float)
    budget = num.array(mysqlfuncs.results_to_list(results,index=3),dtype=float)
    SuccessMetric = misc.returnSuccessMetric(totalGross,budget)

    return releaseYears,title,SuccessMetric,budget,totalGross
コード例 #2
0
ファイル: mysqlfuncs.py プロジェクト: pkundurthy/flopbuster
def get_successMetric_history(person):
    """ get the successMetric and release dates and titles for 
        a given person (also called `part` in moive_meta) """

    # setup the query
    statement1 = 'select title,released,ifnull(totalgross,0),ifnull(budget,0)'
    statement1 += ' from boxoffice where boxoffice.title = any '
    statement1 += '(select `title` from movie_meta '
    statement1 += 'where(part = "%s")) order by released asc;' % (person)

    # get the title, release data, total gross and budget
    results = db_execute_and_fetch(statement1)

    title = results_to_list(results,index=0)
    release = results_to_list(results,index=1)
    totalGross = num.array(results_to_list(results,index=2),dtype=float)
    budget = num.array(results_to_list(results,index=3),dtype=float)

    # compute the sucess metric = log10(totalGross/budget)
    successMetric = misc.returnSuccessMetric(totalGross,budget)

    return release,successMetric,title,budget,totalGross