コード例 #1
0
ファイル: benchmark.py プロジェクト: sxlijin/copenhagent
def nav_branch_factor(shell):
    NAV_LOCS = ['dis', 'noerrebrogade', 'langelinie', 'bryggen']
    NUM_TRIALS = 1000
    LOG_FILENAME = 'nav_average_branching_factor.log'

    result = '%.10f was average branching factor in %d trials at %s.\n' 
    head_pad_len = sum( len(result)
                            - (5 + 2 + 2)
                            + (12 + len(str(NUM_TRIALS)) + len(nav_loc))
                        for nav_loc in NAV_LOCS)

    # open log file for writing
    f = open(LOG_FILENAME, 'w')
    
    tempstr = 'ERROR: average branching factors not calculated due to early exit.'
    f.write(tempstr)
    f.write('_'*(head_pad_len - len(tempstr)))
    f.write('\n')
    f.write('Trial data is included below; format is as follows:\n')
    f.write('avg_b_factor: (vertex, b_factor); (vertex, b_factor); ...)\n')

    f.close()

    overwrite_offset = 0
    for nav_loc in NAV_LOCS:
        # open log file to append
        f = open(LOG_FILENAME, 'a')
        f.write('\nResults for %s below:\n' % nav_loc)
        
        # move agent to $nav_loc
        shell.try_command('map bike locationId=%s' % nav_loc)
        avg_b_factors = []

        for _ in range(NUM_TRIALS):
            # enter navigation and parse FullResponse
            nav_inst = lib.navigation.Instance(shell)
            b_factors = tuple(  (str(v), len(v.get_nexts()))
                                for v in nav_inst.graph.vertex_table.viewvalues()
                                )
            avg_b_factor = average(f[1] for f in b_factors)
            avg_b_factors.append(avg_b_factor)
            f.write('%.6f: ' % avg_b_factor)
            for b_factor in b_factors: f.write(repr(b_factor) + '; ')
            f.write('\n')
            # leave navigation
            shell.try_command('navigation leave')

        f.close()

        # open log file to write at head
        f = open(LOG_FILENAME, 'r+b')
        f.seek(overwrite_offset)
        result_str = result % (average(f[1] for f in b_factors), NUM_TRIALS, nav_loc)
        f.write(result_str)
        overwrite_offset += len(result_str)
        f.close()
コード例 #2
0
ファイル: auton.py プロジェクト: sxlijin/copenhagent
def simple_auton(shell, r, m):
    # hops around navigation activities to win
    agent = shell.active_agent
    while (agent.n_actions < 5000):
        # wait for an activity to become desirable
        while True:
            m.update_seeds(r)
            ((dest, activity), seed) = m.get_best_dest()
            if seed * MULTIPLIERS[activity] > agent.get_avg_creds(): break
            time.sleep(.03)
            r = agent.say('waiting')

        for cmd in m.get_path_from_to(agent.location, dest):
            shell.try_command(cmd)
        r = shell.try_command('%s ai' % activity)