import torch
import torch.nn as nn
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

dataset = pd.read_csv('./train_set.csv')
print(dataset.head())

print(dataset.map_winner)

# teams1 = dataset['team_1'].cat.codes.values
# teams2 = dataset['team_2'].cat.codes.values

print(dataset.dtypes)

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)