entity_results = []
captured_results = []
for trail in range(N_TRIALS):
# Set up the environment
env = Environment(ENV_SIZE, ENV_SIZE, N_PATCHES)
entities = random.randint(MIN_ENTITIES_PER_PATCH, MAX_ENTITIES_PER_PATCH)
for patch in env.children:
patch.create_entities(entities)
s = Searcher(x_pos=(env.length / 2.0),
y_pos=(env.width / 2.0),
parent=env)
for i in range(MAX_MOVES):
s.move()
entity = s.detect()
s.capture(entity)
entity_results.append(entities)
captured_results.append(len(s.captured))
# for k in range(len(entity_results)):
# print(entity_results[k], captured_results[k])
x = np.array(entity_results)
y = np.array(captured_results)
slope, intercept, r_value, p_value, slope_std_error = stats.linregress(x, y)
print "Slope, intercept:", slope, intercept
print "R-squared:", r_value**2
