matchzoo.dataloader.callbacks.dynamic_pooling

Module Contents

class matchzoo.dataloader.callbacks.dynamic_pooling.DynamicPooling(fixed_length_left:int, fixed_length_right:int, compress_ratio_left:float=1, compress_ratio_right:float=1)

Bases: matchzoo.engine.base_callback.BaseCallback

DPoolPairDataGenerator constructor.

Parameters:
  • fixed_length_left – max length of left text.
  • fixed_length_right – max length of right text.
  • compress_ratio_left – the length change ratio, especially after normal pooling layers.
  • compress_ratio_right – the length change ratio, especially after normal pooling layers.
on_batch_unpacked(self, x, y)

Insert dpool_index into x.

Parameters:
  • x – unpacked x.
  • y – unpacked y.
matchzoo.dataloader.callbacks.dynamic_pooling._dynamic_pooling_index(length_left:np.array, length_right:np.array, fixed_length_left:int, fixed_length_right:int, compress_ratio_left:float, compress_ratio_right:float) → np.array