expansion

sCOCA_ML/prepare_data/prepare_gravpot_data.py

@@ -0,0 +1,17 @@
+def prepare_data(batch):
+    
+    phi_ini = batch['input'][:, [1]]
+    D1 = batch['style'][:, [0]]
+    D2 = batch['style'][:, [1]]
+    gravpot = batch['target'][:, [0]]
+    
+    _input = batch['input']
+    _target = (gravpot/D1 - phi_ini)/D1
+    _style = batch['style']
+    
+    return {
+        'input': _input,
+        'target': _target,
+        'style': _style
+    }
+