Spaces:
Sleeping
Sleeping
File size: 1,150 Bytes
9bd4ce5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | import engine_rust
import numpy as np
def analyze_obs():
db_path = "data/cards_compiled.json"
with open(db_path, "r", encoding="utf-8") as f:
db_json = f.read()
db = engine_rust.PyCardDatabase(db_json)
state = engine_rust.PyGameState(db)
# Initialize with a standard deck (ids ~1-60)
# Energy ids ~38-40
p0_deck = [1] * 48 + [101] * 12
p1_deck = [2] * 48 + [102] * 12
state.initialize_game(p0_deck, p1_deck, [38] * 40, [38] * 40, [], [])
obs = state.to_alphazero_tensor()
print(f"Observation Size: {len(obs)}")
print(f"Min: {np.min(obs)}")
print(f"Max: {np.max(obs)}")
print(f"Unique values: {len(np.unique(obs))}")
# Check if int16 can hold it (with scale/offset if needed)
# Range is -32768 to 32767
can_fit_int16 = (np.min(obs) >= -32768) and (np.max(obs) <= 32767)
print(f"Can fit in int16 without scaling? {can_fit_int16}")
# If not, how many unique values actually exist?
# Usually game states are just IDs, counts, flags.
# Large numbers are likely Score/Stats which might need float or scaling.
if __name__ == "__main__":
analyze_obs()
|