runner.FitOrSkipPadding

View source on GitHub

Calculates fit or skip SizeConstraints for GraphTensor padding.

Inherits From: GraphTensorPadding

runner.FitOrSkipPadding(
    gtspec: tfgnn.GraphTensorSpec,
    dataset_provider: runner.DatasetProvider,
    min_nodes_per_component: Optional[Mapping[str, int]] = None,
    fit_or_skip_sample_sample_size: int = 10000,
    fit_or_skip_success_ratio: float = 0.99
)

See: tfgnn.learn_fit_or_skip_size_constraints.

Methods

get_filter_fn

View source

get_filter_fn(
    size_constraints: SizeConstraints
) -> Callable[..., bool]

get_size_constraints

View source

get_size_constraints(
    target_batch_size: int
) -> SizeConstraints