Skip to content

Commit

Permalink
Update detection_evaluation.py
Browse files Browse the repository at this point in the history
  • Loading branch information
thibaultlaugel authored Aug 16, 2019
1 parent 5d351d2 commit bad93e8
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions detection_evaluation.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,14 +63,11 @@ def local_detection_test(obs_to_interprete):
layer = ball[np.where(ball_pred == clf.predict(CF.reshape(1,-1)))]
if k == 0:
ball_ennemies = layer
print(ball_pred.mean())
if (ball_pred.mean() == 1.0) or (ball_pred.mean() == 0.0):
print('cric crac')
N_BALL = 20000
ball = generate_inside_ball_new(obs_to_interprete, (a, b), n=N_BALL)
ball_pred = clf.predict(ball)
if (ball_pred.mean() == 1.0) or (ball_pred.mean() == 0.0):
print('dans ma baraque')
return -2, 1
layer = ball[np.where(ball_pred == clf.predict(CF.reshape(1,-1)))]
else:
Expand Down

0 comments on commit bad93e8

Please sign in to comment.