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

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()

reset_hidden()

forward(tile_matrix1)

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

Bases: torch.nn.Module

init_hidden()

reset_hidden()

forward(pos_tile_matrix, neg_tile_matrix)

test_forward(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()

score(d)

test(d)

zero_grad(*args, **kwargs)