Skip to content
Snippets Groups Projects
Commit 0dc5caff authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fix

parent 9f1bb326
No related merge requests found
Hide whitespace changes
Inline Side-by-side
mhcflurry/train_allele_specific_models_command.py
+ 3
− 3
View file @ 0dc5caff
...@@ -475,12 +475,12 @@ def train_model( ...@@ -475,12 +475,12 @@ def train_model(
pretrain_min_points = hyperparameters['train_data']['pretrain_min_points'] pretrain_min_points = hyperparameters['train_data']['pretrain_min_points']
full_data = GLOBAL_DATA["train_data"] data = GLOBAL_DATA["train_data"]
subset = hyperparameters.get("train_data", {}).get("subset", "all") subset = hyperparameters.get("train_data", {}).get("subset", "all")
if subset == "quantitative": if subset == "quantitative":
data = full_data.loc[ data = data.loc[
full_data.measurement_type == "quantitative" data.measurement_type == "quantitative"
] ]
elif subset == "all": elif subset == "all":
pass pass
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment