capreolus.reranker.DeepTileBar

Module Contents

Classes

DeepTileBar_nn

DeepTileBar_class

DeepTileBar

Zhiwen Tang and Grace Hui Yang. 2019. DeepTileBars: Visualizing Term Distribution for Neural Information Retrieval. In AAAI’19.

Attributes

logger

capreolus.reranker.DeepTileBar.logger[source]
class capreolus.reranker.DeepTileBar.DeepTileBar_nn(p, batch_size, number_filter, lstm_hidden_dim, linear_hidden_dim1, linear_hidden_dim2)[source]

Bases: torch.nn.Module

init_hidden(self)[source]
reset_hidden(self)[source]
forward(self, tile_matrix1)[source]
class capreolus.reranker.DeepTileBar.DeepTileBar_class(extractor, config)[source]

Bases: torch.nn.Module

init_hidden(self)[source]
reset_hidden(self)[source]
forward(self, pos_tile_matrix, neg_tile_matrix)[source]
test_forward(self, pos_tile_matrix)[source]
class capreolus.reranker.DeepTileBar.DeepTileBar(config=None, provide=None, share_dependency_objects=False, build=True)[source]

Bases: capreolus.reranker.Reranker

Zhiwen Tang and Grace Hui Yang. 2019. DeepTileBars: Visualizing Term Distribution for Neural Information Retrieval. In AAAI’19.

module_name = DeepTileBar[source]
dependencies[source]
config_spec[source]
build_model(self)[source]
score(self, d)[source]
test(self, d)[source]
zero_grad(self, *args, **kwargs)[source]