org.campagnelab.dl.somatic.utils.BayesCalibrator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of somatic Show documentation

There is a newer version: 1.4.0

package org.campagnelab.dl.somatic.utils;

import java.io.IOException;

/**
 * Created by rct66 on 7/19/16.
 */
public class BayesCalibrator extends CalcCalibrator {

    public double priorMutRate = (double) 1/(4e6);

    public BayesCalibrator(String modelPath, String prefix, boolean loadStats) throws IOException, ClassNotFoundException {
        super(modelPath, prefix, loadStats);
    }

    @Override
    public float calibrateProb(float prob) {
        double mutGreater = plantedMutSet.subSet(prob,1.1f).size() + 1;
        double unMutGreater = unMutSet.subSet(prob,1.1f).size() + 1;
        double  pMGreater = mutGreater/plantedMutSet.size();
        double  pUGreater = unMutGreater/ unMutSet.size();
        double bayes = pMGreater* priorMutRate /pUGreater;
        return (float) bayes;
    }

    public void setPriorMutRate(double prior){
        priorMutRate = prior;
    }


}