Maxlegrec commited on
Commit
8be66e1
·
verified ·
1 Parent(s): 26b86c7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -1
README.md CHANGED
@@ -44,7 +44,23 @@ model = model.to(device)
44
 
45
  # Get the best move
46
  move = model.get_move_from_fen_no_thinking(fen, T=0.1, device=device)
47
- print(f"Predicted move: {move}")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
48
  ```
49
 
50
  ## Requirements
 
44
 
45
  # Get the best move
46
  move = model.get_move_from_fen_no_thinking(fen, T=0.1, device=device)
47
+ print(f"Policy-based move: {move}")
48
+
49
+ # Get the best move using value
50
+ value_move = model.get_best_move_value(fen, T=0, device=device)
51
+ print(f"Value-based move: {value_move}")
52
+
53
+ # Get position evaluation
54
+ position_value = model.get_position_value(fen, device=device)
55
+ print(f"Position value [black_win, draw, white_win]: {position_value}")
56
+
57
+ # Get move probabilities
58
+ probs = model.get_move_from_fen_no_thinking(fen, T=1, device=device, return_probs=True)
59
+ top_moves = sorted(probs.items(), key=lambda x: x[1], reverse=True)[:5]
60
+ print("Top 5 moves:")
61
+ for move, prob in top_moves:
62
+ print(f" {move}: {prob:.4f}")
63
+
64
  ```
65
 
66
  ## Requirements