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

add widelocal variant

parent f21d9ebb
No related merge requests found
...@@ -34,10 +34,10 @@ time mhcflurry-class1-train-allele-specific-models \ ...@@ -34,10 +34,10 @@ time mhcflurry-class1-train-allele-specific-models \
--hyperparameters hyperparameters-standard.yaml \ --hyperparameters hyperparameters-standard.yaml \
--out-models-dir models-standard-quantitative \ --out-models-dir models-standard-quantitative \
--percent-rank-calibration-num-peptides-per-length 0 \ --percent-rank-calibration-num-peptides-per-length 0 \
--alleles $ALLELES & --allele $ALLELES 2>&1 | tee -a LOG.standard.txt &
# Model variations on qualitative + quantitative # Model variations on qualitative + quantitative
for mod in 0local_noL1 0local 2local dense16 dense64 noL1 onehot embedding for mod in 0local_noL1 0local 2local widelocal dense16 dense64 noL1 onehot embedding
do do
cp $SCRIPT_DIR/hyperparameters-${mod}.yaml . cp $SCRIPT_DIR/hyperparameters-${mod}.yaml .
mkdir models-${mod} mkdir models-${mod}
...@@ -46,12 +46,15 @@ do ...@@ -46,12 +46,15 @@ do
--hyperparameters hyperparameters-${mod}.yaml \ --hyperparameters hyperparameters-${mod}.yaml \
--out-models-dir models-${mod} \ --out-models-dir models-${mod} \
--percent-rank-calibration-num-peptides-per-length 0 \ --percent-rank-calibration-num-peptides-per-length 0 \
--alleles $ALLELES & --allele $ALLELES 2>&1 | tee -a LOG.${mod}.txt &
done done
wait wait
cp $SCRIPT_ABSOLUTE_PATH . cp $SCRIPT_ABSOLUTE_PATH .
bzip2 LOG.txt for i in $(ls *.txt)
do
bzip2 $i
done
tar -cjf "../${DOWNLOAD_NAME}.tar.bz2" * tar -cjf "../${DOWNLOAD_NAME}.tar.bz2" *
echo "Created archive: $SCRATCH_DIR/$DOWNLOAD_NAME.tar.bz2" echo "Created archive: $SCRATCH_DIR/$DOWNLOAD_NAME.tar.bz2"
[{
##########################################
# ENSEMBLE SIZE
##########################################
"n_models": 8,
##########################################
# OPTIMIZATION
##########################################
"max_epochs": 500,
"patience": 10,
"early_stopping": true,
"validation_split": 0.2,
##########################################
# RANDOM NEGATIVE PEPTIDES
##########################################
"random_negative_rate": 0.0,
"random_negative_constant": 25,
"random_negative_affinity_min": 20000.0,
"random_negative_affinity_max": 50000.0,
##########################################
# PEPTIDE REPRESENTATION
##########################################
# One of "one-hot", "embedding", or "BLOSUM62".
"peptide_amino_acid_encoding": "BLOSUM62",
"use_embedding": false, # maintained for backward compatability
"embedding_output_dim": 8, # only used if using embedding
"kmer_size": 15,
##########################################
# NEURAL NETWORK ARCHITECTURE
##########################################
"locally_connected_layers": [
{
"filters": 8,
"activation": "tanh",
"kernel_size": 5
}
],
"activation": "relu",
"output_activation": "sigmoid",
"layer_sizes": [
32
],
"dense_layer_l1_regularization": 0.001,
"batch_normalization": false,
"dropout_probability": 0.0,
}]
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment