rsmillie94 commited on
Commit
fc0cdf7
·
verified ·
1 Parent(s): aa828d3

Task 18 Dark Pool Model

Browse files
Files changed (3) hide show
  1. README.md +60 -0
  2. model.onnx +3 -0
  3. model.pt +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:8403cda49256bec482654b17380ee811f9a4910f0da1cb434d8be7fac2733648
3
+ size 11655
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48b3c86a5484af18bd9fcb616850da03aa71d8dd30f55771ef48cc3c2e3dd0b8
3
+ size 13321