trading-models

Intro

Neural Network models for trading: MLP, CNN, Transformer
Below shows the training process of my private model

Usage

pip install trading-models

import torch as tc
from torch.utils.data import DataLoader

from trading_models.simple_models import CNN, MLP, Transformer
from trading_models.utils import WindowDataset, model_size

"""
trading models

input: x.shape = (T, F)
output: a.shape = (T-W+1, A)

T: time
F: features
W: window length
A: actions

at each time step t in range(W-1, T),
the model looks at data in the window
x[t+1-W : t+1, :]
to make A actions
"""

T, F, W, A = 100, 2, 50, 1
device = tc.device("cuda" if tc.cuda.is_available() else "cpu")
x = tc.randn(T, F).to(device)

dataset = WindowDataset(x, W)
dataloader = DataLoader(dataset, batch_size=32, shuffle=False)

net1 = MLP([W * F, 64, 64, A]).to(device)
net2 = CNN([F, 64, 64, A]).to(device)
net3 = Transformer(W, F, A, d_model=64, d_ff=64, n_head=2, n_layer=2).to(device)

for net in [net1, net2, net3]:
    outputs = []
    for batch in dataloader:
        outputs.append(net(batch.to(device)).detach())
    output = tc.cat(outputs, dim=0)
    print(net, output.shape, model_size(net), "\n")

Output

MLP(
  (mlp): Sequential(
    (0): Linear(in_features=100, out_features=64, bias=True)
    (1): SiLU()
    (2): Linear(in_features=64, out_features=64, bias=True)
    (3): SiLU()
    (4): Linear(in_features=64, out_features=1, bias=True)
  )
) torch.Size([51, 1]) trainable: 10689/10689 

CNN(
  (cnn): Sequential(
    (0): Conv1d(2, 64, kernel_size=(3,), stride=(1,), padding=(1,))
    (1): SiLU()
    (2): Conv1d(64, 64, kernel_size=(3,), stride=(1,), padding=(1,))
    (3): SiLU()
    (4): AdaptiveAvgPool1d(output_size=1)
  )
  (fc): Linear(in_features=64, out_features=1, bias=True)
) torch.Size([51, 1]) trainable: 12865/12865 

Transformer(
  (proj): Linear(in_features=2, out_features=64, bias=True)
  (trans): TransformerEncoder(
    (layers): ModuleList(
      (0-1): 2 x TransformerEncoderLayer(
        (self_attn): MultiheadAttention(
          (out_proj): NonDynamicallyQuantizableLinear(in_features=64, out_features=64, bias=True)
        )
        (linear1): Linear(in_features=64, out_features=64, bias=True)
        (dropout): Dropout(p=0.1, inplace=False)
        (linear2): Linear(in_features=64, out_features=64, bias=True)
        (norm1): LayerNorm((64,), eps=1e-05, elementwise_affine=True)
        (norm2): LayerNorm((64,), eps=1e-05, elementwise_affine=True)
        (dropout1): Dropout(p=0.1, inplace=False)
        (dropout2): Dropout(p=0.1, inplace=False)
      )
    )
  )
  (fc): Linear(in_features=64, out_features=1, bias=True)
) torch.Size([51, 1]) trainable: 50689/50689

Metadata

Release files for trading-models 0.3.7

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for trading-models 0.3.7
File	Size	Uploaded
trading_models-0.3.7.tar.gz	10.7 kB	Jan 19, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for trading-models 0.3.7
File	Interpreter	ABI	Platform	Reset
trading_models-0.3.7-py3-none-any.whl 10.7 kB Jan 19, 2026	Python 3	none	any	Details

Total release size: 21.4 kB

Release files / trading_models-0.3.7.tar.gz

Download URL	trading_models-0.3.7.tar.gz
Size	10.7 kB
Tags	Source
SHA-256 checksum How to use checksums	`753a4d7fd4116b715fb9ce34f584b91818ae98a5cbe7b82a3a09728354918369`
BLAKE2b-256 checksum How to use checksums	`4c85926eb87237a3d33d25e1dcf7d90038f82464b1ac77ddb1130e455323e50a`
Upload date	Jan 19, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.2.0 CPython/3.11.14`

Release files / trading_models-0.3.7-py3-none-any.whl

Download URL	trading_models-0.3.7-py3-none-any.whl
Size	10.7 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`a9262496a3c99a940d79724f2255b6dc6b8296255e55fb049a6e72fe2e9670ce`
BLAKE2b-256 checksum How to use checksums	`bf9ee3aed361112836c7bcf65989abd21d962ed0b8e75415538369b38165c210`
Upload date	Jan 19, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.2.0 CPython/3.11.14`