capreolus.reranker.DeepTileBar

Module Contents

Classes

DeepTileBar_nn(p, batch_size, number_filter, lstm_hidden_dim, linear_hidden_dim1, linear_hidden_dim2)
DeepTileBar_class(extractor, config)
DeepTileBar(config=None, provide=None, share_dependency_objects=False, build=True)	Zhiwen Tang and Grace Hui Yang. 2019. DeepTileBars: Visualizing Term Distribution for Neural Information Retrieval. In AAAI‘19.

capreolus.reranker.DeepTileBar.logger

class capreolus.reranker.DeepTileBar.DeepTileBar_nn(p, batch_size, number_filter, lstm_hidden_dim, linear_hidden_dim1, linear_hidden_dim2)

Bases: torch.nn.Module

init_hidden(self)

reset_hidden(self)

forward(self, tile_matrix1)

class capreolus.reranker.DeepTileBar.DeepTileBar_class(extractor, config)

Bases: torch.nn.Module

init_hidden(self)

reset_hidden(self)

forward(self, pos_tile_matrix, neg_tile_matrix)

test_forward(self, pos_tile_matrix)

class capreolus.reranker.DeepTileBar.DeepTileBar(config=None, provide=None, share_dependency_objects=False, build=True)

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

dependencies

config_spec

build_model(self)

score(self, d)

test(self, d)

zero_grad(self, *args, **kwargs)