univi.config.TokenizerConfig

class univi.config.TokenizerConfig(mode: "Literal['topk_scalar', 'topk_channels', 'patch', 'topk_embed']" = 'topk_scalar', n_tokens: 'int' = 256, channels: "Sequence[Literal['value', 'rank', 'dropout']]" = ('value',), patch_size: 'int' = 32, patch_proj_dim: 'Optional[int]' = None, add_cls_token: 'bool' = False, n_features: 'Optional[int]' = None, d_model: 'Optional[int]' = None, value_mlp_hidden: 'int' = 256, use_coords: 'bool' = False, chrom_vocab_size: 'int' = 0, coord_scale: 'float' = 1000000.0, feature_info: 'Optional[Dict[str, Any]]' = None, use_feature_embedding: 'bool' = False, feature_emb_dim: 'Optional[int]' = None, feature_emb_mode: "Literal['add', 'concat']" = 'add', use_coord_embedding: 'bool' = False, n_chroms: 'Optional[int]' = None, coord_emb_dim: 'Optional[int]' = None, coord_mode: "Literal['midpoint', 'interval']" = 'midpoint', coord_mlp_hidden: 'int' = 128, coord_num_frequencies: 'int' = 16, token_proj_dim: 'Optional[int]' = None)

Bases: object

__init__(mode='topk_scalar', n_tokens=256, channels=('value',), patch_size=32, patch_proj_dim=None, add_cls_token=False, n_features=None, d_model=None, value_mlp_hidden=256, use_coords=False, chrom_vocab_size=0, coord_scale=1000000.0, feature_info=None, use_feature_embedding=False, feature_emb_dim=None, feature_emb_mode='add', use_coord_embedding=False, n_chroms=None, coord_emb_dim=None, coord_mode='midpoint', coord_mlp_hidden=128, coord_num_frequencies=16, token_proj_dim=None)

Methods

__init__([mode, n_tokens, channels, ...])

Attributes

add_cls_token
channels
chrom_vocab_size
coord_emb_dim
coord_mlp_hidden
coord_mode
coord_num_frequencies
coord_scale
d_model
feature_emb_dim
feature_emb_mode
feature_info
mode
n_chroms
n_features
n_tokens
patch_proj_dim
patch_size
token_proj_dim
use_coord_embedding
use_coords
use_feature_embedding
value_mlp_hidden