henri/trictrac
1
1
Fork 0
Code Issues 13 Pull requests Projects 1 Releases Packages Wiki Activity Actions

feat: add get_tensor on GameState more explicit for training than the minimal get_vec()

Browse source
This commit is contained in:
Henri Bourcereau 2026-03-07 12:56:03 +01:00
parent 145ab7dcda
commit aa7f5fe42a
4 changed files with 108 additions and 1000 deletions
Download patch file Download diff file Expand all files Collapse all files

6
store/src/pyengine.rs
View file

@ -113,11 +113,11 @@ impl TricTrac {
[self.get_score(1), self.get_score(2)]
}
fn get_tensor(&self, player_idx: u64) -> Vec<i8> {
fn get_tensor(&self, player_idx: u64) -> Vec<f32> {
if player_idx == 0 {
self.game_state.to_vec()
self.game_state.to_tensor()
} else {
self.game_state.mirror().to_vec()
self.game_state.mirror().to_tensor()
}
}