From 1d8fbac1352451fd2e6cd7bfbe6aafc8e1f94537 Mon Sep 17 00:00:00 2001 From: Neil Ferguson Date: Tue, 30 Jun 2015 21:40:13 +0100 Subject: [PATCH] Added more population groups --- src/main/scala/com/neilferguson/PopStrat.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/main/scala/com/neilferguson/PopStrat.scala b/src/main/scala/com/neilferguson/PopStrat.scala index ba33e50..5ab5ec7 100644 --- a/src/main/scala/com/neilferguson/PopStrat.scala +++ b/src/main/scala/com/neilferguson/PopStrat.scala @@ -31,7 +31,7 @@ object PopStrat { val sc = new SparkContext(conf) // Create a set of the populations that we want to predict - val populations = Set("GBR", "ASW", "CHB") + val populations = Set("ACB", "ASW", "BEB", "CDX", "CEU", "CHB", "CHS", "CLM", "ESN", "FIN", "GBR", "GIH") // Create a map of sample ID -> population so that we can filter out the samples we're not interested in val panel: Map[String,String] = extract(panelFile, (sampleID: String, pop: String) => populations.contains(pop)) @@ -116,7 +116,7 @@ object PopStrat { deepLearningParameters._train = training deepLearningParameters._valid = test deepLearningParameters._response_column = "Region" - deepLearningParameters._epochs = 10 + deepLearningParameters._epochs = 50 deepLearningParameters._activation = Activation.RectifierWithDropout deepLearningParameters._hidden = Array[Int](100,100)