target_extraction.allen.modules package

Subpackages

• target_extraction.allen.modules.inter_target package
  • Submodules
  • target_extraction.allen.modules.inter_target.inter_target module
  • target_extraction.allen.modules.inter_target.sequence_inter_target module
  • Module contents
• target_extraction.allen.modules.target_position_weight package
  • Submodules
  • target_extraction.allen.modules.target_position_weight.relative_target_position_weight module
  • target_extraction.allen.modules.target_position_weight.target_position_weight module
  • Module contents

Submodules

target_extraction.allen.modules.word_dropout module

class target_extraction.allen.modules.word_dropout.WordDrouput(p)

Bases: torch.nn.modules.module.Module

Word Dropout will randomly drop whole words/timesteps. This is equivalent to `1D Spatial Dropout`_.

forward(embedded_text)

Parameters

embedded_text (FloatTensor) – A tensor of shape: [batch_size, timestep, embedding_dim] of which the dropout will drop entire timestep which is the equivalent to words.

Return type

FloatTensor

Returns

The given tensor but with timesteps/words dropped.

Module contents