matchzoo.dataloader.callbacks.histogram
¶
Module Contents¶
-
class
matchzoo.dataloader.callbacks.histogram.
Histogram
(embedding_matrix:np.ndarray, bin_size:int=30, hist_mode:str='CH')¶ Bases:
matchzoo.engine.base_callback.BaseCallback
Generate data with matching histogram.
Parameters: - embedding_matrix – The embedding matrix used to generator match histogram.
- bin_size – The number of bin size of the histogram.
- hist_mode – The mode of the
MatchingHistogramUnit
, one of CH, NH, and LCH.
-
on_batch_unpacked
(self, x, y)¶ Insert match_histogram to x.
-
matchzoo.dataloader.callbacks.histogram.
_trunc_text
(input_text:list, length:list) → list¶ Truncating the input text according to the input length.
Parameters: - input_text – The input text need to be truncated.
- length – The length used to truncated the text.
Returns: The truncated text.
-
matchzoo.dataloader.callbacks.histogram.
_build_match_histogram
(x:dict, match_hist_unit:mz.preprocessors.units.MatchingHistogram) → np.ndarray¶ Generate the matching hisogram for input.
Parameters: - x – The input dict.
- match_hist_unit – The histogram unit
MatchingHistogramUnit
.
Returns: The matching histogram.