spawnrys/flanabot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve Connect4Bot ai

Browse Source
This commit is contained in:
AlberLC
2022-11-22 00:41:08 +01:00
parent c3ab1be08e
commit f0babe7c85
1 changed files with 1 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
flanabot/bots/connect_4_bot.py
View File

@@ -33,18 +33,9 @@ class Connect4Bot(MultiBot, ABC):
board = message.contents['connect_4']['board']
player_1 = Player.from_dict(message.contents['connect_4']['player_1'])
player_2 = Player.from_dict(message.contents['connect_4']['player_2'])
turn = message.contents['connect_4']['turn']
available_positions_ = self._available_positions(board)
# first move forced to center
if turn <= 1:
j = constants.CONNECT_4_N_COLUMNS // 2
if (constants.CONNECT_4_N_ROWS - 1, j) in available_positions_:
return self.insert_piece(j, player_2.number, message)
else:
return self.insert_piece(random.choice((j - 1, j + 1)), player_2.number, message)
# check if ai can win
for i, j in available_positions_:
if player_2.number in self._check_winners(i, j, board):
@@ -94,10 +85,7 @@ class Connect4Bot(MultiBot, ABC):
good_positions = [pos for pos in available_positions_ if pos not in human_winning_positions_above and pos not in ai_winning_positions_above]
if good_positions:
if random.random() < 0.5:
j = random.choice(self._best_plays(good_positions, player_2.number, board))[1]
else:
j = random.choice(good_positions)[1]
j = random.choice(self._best_plays(good_positions, player_2.number, board))[1]
elif ai_winning_positions_above:
j = random.choice(self._best_plays(ai_winning_positions_above, player_2.number, board))[1]
else: