Gitdeeper4 commited on
Commit
142d6ca
·
verified ·
1 Parent(s): 1917451

Upload chi_model.py

Browse files
Files changed (1) hide show
  1. chi_model.py +28 -0
chi_model.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # TSU-WAVE CHI Model
2
+ import numpy as np
3
+
4
+ def predict_chi(wcc, kpr, hfsi, becf, sdb, sbsp, smvi):
5
+ """Simple CHI prediction model"""
6
+ weights = {
7
+ 'wcc': 0.12, 'kpr': 0.19, 'hfsi': 0.24,
8
+ 'becf': 0.21, 'sdb': 0.08, 'sbsp': 0.11, 'smvi': 0.05
9
+ }
10
+
11
+ # Normalize parameters (simplified)
12
+ params = {
13
+ 'wcc': min(wcc/1.58, 1.0),
14
+ 'kpr': min(kpr/2.0, 1.0),
15
+ 'hfsi': 1 - min(hfsi/1.0, 1.0),
16
+ 'becf': min(becf/6.0, 1.0),
17
+ 'sdb': 1 - min(sdb/3.5, 1.0),
18
+ 'sbsp': min(sbsp/1.2, 1.0),
19
+ 'smvi': min(smvi/0.6, 1.0)
20
+ }
21
+
22
+ chi = sum(weights[p] * params[p] for p in weights)
23
+ return chi
24
+
25
+ if __name__ == '__main__':
26
+ # Tōhoku 2011 example
27
+ chi = predict_chi(1.56, 1.89, 0.31, 7.3, 0.8, 1.18, 0.38)
28
+ print(f'CHI: {chi:.3f}')