From f91a85cf52b8ab43902b5f4a974ba557e97e278c Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sat, 25 Nov 2017 13:54:55 -0500
Subject: [PATCH] new architectures
---
downloads-generation/models_class1/GENERATE.sh | 2 +-
downloads-generation/models_class1/hyperparameters.yaml | 7 +------
2 files changed, 2 insertions(+), 7 deletions(-)
diff --git a/downloads-generation/models_class1/GENERATE.sh b/downloads-generation/models_class1/GENERATE.sh
index ae98f51e..a38faa48 100755
--- a/downloads-generation/models_class1/GENERATE.sh
+++ b/downloads-generation/models_class1/GENERATE.sh
@@ -31,7 +31,7 @@ time mhcflurry-class1-train-allele-specific-models \
--data "$(mhcflurry-downloads path data_curated)/curated_training_data.csv.bz2" \
--hyperparameters hyperparameters.yaml \
--out-models-dir models \
- --min-measurements-per-allele 200
+ --min-measurements-per-allele 100
cp $SCRIPT_ABSOLUTE_PATH .
bzip2 LOG.txt
diff --git a/downloads-generation/models_class1/hyperparameters.yaml b/downloads-generation/models_class1/hyperparameters.yaml
index 0e114b32..30b78fa2 100644
--- a/downloads-generation/models_class1/hyperparameters.yaml
+++ b/downloads-generation/models_class1/hyperparameters.yaml
@@ -2,7 +2,7 @@
##########################################
# ENSEMBLE SIZE
##########################################
-"n_models": 12,
+"n_models": 8,
##########################################
# OPTIMIZATION
@@ -32,11 +32,6 @@
# NEURAL NETWORK ARCHITECTURE
##########################################
"locally_connected_layers": [
- {
- "filters": 8,
- "activation": "tanh",
- "kernel_size": 3
- },
{
"filters": 8,
"activation": "tanh",
--
GitLab