calibration working as expected

tiger.py

@@ -123,6 +123,13 @@ def process_data(transcript_seq: str):
     return target_seq, guide_seq, model_inputs
 
 
+def calibrate_predictions(predictions: np.array, num_mismatches: np.array, params: pd.DataFrame = None):
+    if params is None:
+        params = pd.read_pickle('calibration_params.pkl')
+    correction = np.squeeze(params.set_index('num_mismatches').loc[num_mismatches].to_numpy())
+    return correction * predictions
+
+
 def prediction_transform(predictions: np.array, **params):
     if len(params) == 0:
         with open('transform_params.pkl', 'rb') as f: