Spaces:
Runtime error
Runtime error
File size: 1,443 Bytes
cb8fc87 97794a7 9115421 cb8fc87 9115421 cb8fc87 9115421 cb8fc87 9115421 6ff217d cb8fc87 6ff217d 9115421 97794a7 9115421 97794a7 9115421 97794a7 9115421 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import chess
import matplotlib.pyplot as plt
import pandas as pd
from typing import Dict
from logger import Logger
class GameDatabase:
def __init__(self):
self.db = []
pass
def add_game(self, game: Dict[str, str]):
self.db.append(game)
def display_game(self, game_num: int): #Displays analytics for a specific game
pass
def display_tournament(self): #Displays analytics for the entire tournament
df = pd.DataFrame(self.db)
#heatmap of tournament winrates
#bar chart of tournament winrates
win_results = df["Winner"].value_counts()
print(win_results.rank())
win_results.plot.bar()
plt.show()
pass
if __name__ == "__main__":
test_logger = Logger("ChessGPT", "ChatGPT")
test_logger.add_cheat("ChessGPT")
test_logger.add_legal_move("e4")
test_logger.add_checkmate("ChessGPT")
#test_logger.add_legal_move("e4 e5")
formatted = test_logger.return_formatted_game()
test_logger_2 = Logger("ChessGPT", "BERT")
test_logger_2.add_checkmate("BERT")
formatted_2 = test_logger_2.return_formatted_game()
test_logger_3 = Logger("ChessGPT", "BERT")
test_logger_3.add_checkmate("ChessGPT")
formatted_3 = test_logger_3.return_formatted_game()
db = GameDatabase()
db.add_game(formatted)
db.add_game(formatted_2)
db.add_game(formatted_3)
db.display_tournament()
|