capreolus.reranker.base

Module Contents

Classes

Reranker()
class capreolus.reranker.base.Reranker[source]

Bases: profane.ModuleBase

module_type = reranker[source]
dependencies[source]
add_summary(self, summary_writer, niter)[source]

Write to the summay_writer custom visualizations/data specific to this reranker

save_weights(self, weights_fn, optimizer)[source]
load_weights(self, weights_fn, optimizer)[source]