popov/gogo2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

multi pair inputs, wip, working training??

Browse Source
This commit is contained in:
Dobromir Popov
2025-03-29 01:56:45 +02:00
parent b313f70cd7
commit 0b2000e3e7
9 changed files with 686 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
NN/models/__init__.py
View File

@ -3,12 +3,17 @@ Neural Network Models
====================
This package contains the neural network models used in the trading system:
- CNN Model: Deep convolutional neural network for feature extraction
- Transformer Model: Processes high-level features for improved pattern recognition
- CNN Model: Deep convolutional neural network for feature extraction
- Transformer Model: Processes high-level features for improved pattern recognition
- MoE: Mixture of Experts model that combines multiple neural networks
PyTorch implementation only.
"""
from NN.models.cnn_model import CNNModel
from NN.models.transformer_model import TransformerModel, TransformerBlock, MixtureOfExpertsModel
from NN.models.cnn_model_pytorch import CNNModelPyTorch as CNNModel
from NN.models.transformer_model_pytorch import (
TransformerModelPyTorch as TransformerModel,
MixtureOfExpertsModelPyTorch as MixtureOfExpertsModel
)
__all__ = ['CNNModel', 'TransformerModel', 'TransformerBlock', 'MixtureOfExpertsModel']
__all__ = ['CNNModel', 'TransformerModel', 'MixtureOfExpertsModel']