File size: 938 Bytes
ee6da62 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | """
TD3B: Transition-Directed Discrete Diffusion for Binders
A module extending TR2-D2 with directional allosteric control.
"""
from .direction_oracle import DirectionalOracle
from .td3b_scoring import TD3BRewardFunction, TD3BConfidenceWeighting, create_td3b_reward_function
from .td3b_losses import ContrastiveLoss, InfoNCELoss, TD3BTotalLoss, extract_embeddings_from_mdlm
from .td3b_mcts import TD3B_MCTS, create_td3b_mcts
from .td3b_finetune import td3b_finetune, add_td3b_sampling_to_model
from .data_utils import TD3BDataset, load_td3b_data
__all__ = [
'DirectionalOracle',
'TD3BRewardFunction',
'TD3BConfidenceWeighting',
'create_td3b_reward_function',
'ContrastiveLoss',
'InfoNCELoss',
'TD3BTotalLoss',
'extract_embeddings_from_mdlm',
'TD3B_MCTS',
'create_td3b_mcts',
'td3b_finetune',
'add_td3b_sampling_to_model',
'TD3BDataset',
'load_td3b_data',
]
__version__ = '0.1.0'
|