(cards)
| 64 | net.load_state_dict(torch.load('./bid_weights.pkl', map_location=torch.device("cpu"))) |
| 65 | |
| 66 | def predict(cards): |
| 67 | input = RealToOnehot(cards) |
| 68 | if UseGPU: |
| 69 | input = input.to(device) |
| 70 | input = torch.flatten(input) |
| 71 | win_rate = net(input) |
| 72 | return win_rate[0].item() * 100 |
| 73 | |
| 74 | def predict_env(cards): |
| 75 | input = EnvToOnehot(cards) |
nothing calls this directly
no test coverage detected