20 lines
422 B
Python
20 lines
422 B
Python
|
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)
|