dataset.py File Reference

Classes
class	dataset.ChessMoEFactorizedDataset
class	dataset.ChessExpertFactorizedDataset

Namespaces
namespace	dataset

Functions
	dataset.decode_feature_row (row, feature_layout, planes_per_type, packed_offsets, packed_branch_planes, num_bypass_planes)

Variables
list	dataset.PLANES_PER_GROUP_CONST = [4, 4, 5, 5, 5, 4, 4, 4, 5, 5, 5, 4]
list	dataset.PACKED_OFFSETS_CONST
	dataset.parser = argparse.ArgumentParser()
	dataset.help
	dataset.type
	dataset.int
	dataset.default
	dataset.args = parser.parse_args()
	dataset.ds = ChessMoEFactorizedDataset(args.data_dir, max_samples=args.max, n_globals=args.n_globals)
	dataset.sample = ds[0]