rsmillie94 commited on
Commit
0ff4d31
·
verified ·
1 Parent(s): 3750b95

Task 18 Dark Pool Model

Browse files
Files changed (4) hide show
  1. README.md +60 -0
  2. model.onnx +3 -0
  3. model.pt +3 -0
  4. model_log.npz +3 -0
README.md ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - onnx
5
+ - flock
6
+ - dark-pool
7
+ - trading
8
+ - reinforcement-learning
9
+ ---
10
+
11
+ # Flock.io Task 18: Dark Pool Trading Model
12
+
13
+ This is a neural network model trained for the Flock.io Task 18 - Dark Pool Trading.
14
+
15
+ ## Model Details
16
+
17
+ - **Task**: Dark Pool Trading Prediction
18
+ - **Framework**: PyTorch → ONNX
19
+ - **Input**: 34 features (market state)
20
+ - **Output**: 1 value (predicted fill rate / action value)
21
+ - **Parameters**: ~1.78M (under 3M limit)
22
+
23
+ ## Architecture
24
+
25
+ Multi-Layer Perceptron (MLP):
26
+ - Input: 34 features
27
+ - Hidden layers: 1024 → 1024 → 512 → 256 → 128
28
+ - Output: 1 value
29
+ - Activation: ReLU
30
+ - Normalization: BatchNorm
31
+ - Regularization: Dropout (0.2)
32
+
33
+ ## Usage
34
+
35
+ ```python
36
+ import onnxruntime as ort
37
+ import numpy as np
38
+
39
+ # Load the model
40
+ session = ort.InferenceSession("model.onnx")
41
+
42
+ # Prepare input (34 features)
43
+ input_data = np.random.randn(1, 34).astype(np.float32)
44
+
45
+ # Run inference
46
+ outputs = session.run(None, {"input": input_data})
47
+ prediction = outputs[0]
48
+ print(f"Prediction: {prediction}")
49
+ ```
50
+
51
+ ## Training
52
+
53
+ Trained on Flock.io Task 18 dataset:
54
+ - Training samples: 1200
55
+ - Validation samples: 400
56
+ - Best validation loss: ~0.001
57
+
58
+ ## License
59
+
60
+ Apache 2.0
model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a186af29946b1c8fc12d2ed6020a72fd64a15be63439b1145ab0d63b01149ac
3
+ size 97762
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9017992b797074e42c92eb16e7d5942c83a624de6e351660866b3b696860f2d
3
+ size 70746
model_log.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3feb0a91ca0276aa6846a67557a440b321916b2a524bd0041a5c007964923b07
3
+ size 1634