def save_json_strelka_format(self, filename, varianttype, threshold):
""" Save to json including all strelka scoring model meta-data """
import datetime
import json
date = datetime.datetime.utcnow().isoformat()
meta = {
"Date": "%sZ" % (date),
"Features": self.clf.columns,
"ModelType": "RandomForest",
"FilterCutoff": threshold,
"Calibration": {
"Power": 1,
"Scale": 1
}
}
all_trees = io.classifier_to_dict(self.clf)
full_model = meta
full_model["Model"] = all_trees
modelFile = {
"CalibrationModels": {
"Somatic": {
varianttype: full_model
}
}
}
json.dump(modelFile, open(filename, "wb"))
def save_json_strelka_format(self, filename):
""" Save to json including all strelka scoring model meta-data """
import datetime
import json
date = datetime.datetime.utcnow().isoformat()
meta = {
"Date": "%sZ" % (date),
"Features": self.clf.columns,
"ModelType": "RandomForest",
"FilterCutoff": 0.5
}
all_trees = io.classifier_to_dict(self.clf)
#full_model = {"Meta" : meta, "Model" : all_trees }
full_model = meta
full_model["Model"] = all_trees
modelFile = {"CalibrationModels": {"Germline": {"SNP": full_model}}}
json.dump(modelFile, open(filename, "wb"))