Andrew Stirn commited on
Commit
0b0f412
·
1 Parent(s): 0590665

before and after calibration checks out

Browse files
Files changed (2) hide show
  1. calibration_params.pkl +0 -0
  2. tiger.py +1 -1
calibration_params.pkl CHANGED
Binary files a/calibration_params.pkl and b/calibration_params.pkl differ
 
tiger.py CHANGED
@@ -126,7 +126,7 @@ def process_data(transcript_seq: str):
126
  def calibrate_predictions(predictions: np.array, num_mismatches: np.array, params: pd.DataFrame = None):
127
  if params is None:
128
  params = pd.read_pickle('calibration_params.pkl')
129
- correction = np.squeeze(params.set_index('num_mismatches').loc[num_mismatches].to_numpy())
130
  return correction * predictions
131
 
132
 
 
126
  def calibrate_predictions(predictions: np.array, num_mismatches: np.array, params: pd.DataFrame = None):
127
  if params is None:
128
  params = pd.read_pickle('calibration_params.pkl')
129
+ correction = np.squeeze(params.set_index('num_mismatches').loc[num_mismatches, 'slope'].to_numpy())
130
  return correction * predictions
131
 
132