PositionSplit.position_keys

property PositionSplit.position_keys: list[str]

Position keys contained in this split.

Examples

>>> import jax.numpy as jnp
>>> from liesel.optim import PositionSplit
>>> from liesel.optim.types import Position
>>> split = PositionSplit(
...     Position({"x": jnp.arange(2)}),
...     Position({}),
...     Position({}),
...     train_axis_size=2,
...     validate_axis_size=0,
...     test_axis_size=0,
... )
>>> split.position_keys
['x']