velocitatem/PHANTOM
1
0
Fork 0
mirror of https://github.com/velocitatem/PHANTOM.git synced 2026-05-31 16:43:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

acapting some architectures

Browse Source
This commit is contained in:
Daniel Rosel
2026-01-21 18:22:39 +01:00
parent 6aad196234
commit 5f607a58eb
2 changed files with 247 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
experiments/ml/__init__.py
View File

@@ -1,11 +1,21 @@
from .evals import evaluate
from .arch import (
XGBoostAgentClassifier,
LightGBMAgentClassifier
LightGBMAgentClassifier,
ContrastiveWeakClassifier,
TrajectoryEncoder,
WeakClassifier,
contrastive_loss,
featurize_trajectory,
)
__all__ =[
__all__ = [
'evaluate',
'XGBoostAgentClassifier',
'LightGBMAgentClassifier'
'LightGBMAgentClassifier',
'ContrastiveWeakClassifier',
'TrajectoryEncoder',
'WeakClassifier',
'contrastive_loss',
'featurize_trajectory',
]