tfma.size_estimator.SizeEstimator

Size estimator.

tfma.size_estimator.SizeEstimator(
    size_threshold: int,
    size_fn: Callable[[Any], int]
)

Methods

clear

View source

clear()

get_estimate

View source

get_estimate() -> int

should_flush

View source

should_flush() -> bool

update

View source

update(
    value: Any
)

Update current size based on the input value.