Release/0.2 (#2)

* models * longer names * rename to v0.2 --------- Co-authored-by: Will Dumm <[email protected]>
matsengrp · Nov 13, 2024 · ba43b42 · ba43b42
1 parent 50cef92
commit ba43b42
Show file tree

Hide file tree

Showing 6 changed files with 48 additions and 0 deletions.
diff --git a/models/ThriftyHumV0.2-20.pth b/models/ThriftyHumV0.2-20.pth
diff --git a/models/ThriftyHumV0.2-20.yml b/models/ThriftyHumV0.2-20.yml
@@ -0,0 +1,16 @@
+encoder_class: KmerSequenceEncoder
+encoder_parameters:
+  kmer_length: 3
+  site_count: 500
+model_class: IndepRSCNNModel
+model_hyperparameters:
+  dropout_prob: 0.3
+  embedding_dim: 7
+  filter_count: 19
+  kernel_size: 11
+  kmer_length: 3
+serialization_version: 0
+training_hyperparameters:
+  learning_rate: 0.001
+  min_learning_rate: 1.0e-06
+  weight_decay: 1.0e-06
diff --git a/models/ThriftyHumV0.2-45.pth b/models/ThriftyHumV0.2-45.pth
diff --git a/models/ThriftyHumV0.2-45.yml b/models/ThriftyHumV0.2-45.yml
@@ -0,0 +1,16 @@
+encoder_class: KmerSequenceEncoder
+encoder_parameters:
+  kmer_length: 3
+  site_count: 500
+model_class: IndepRSCNNModel
+model_hyperparameters:
+  dropout_prob: 0.2
+  embedding_dim: 7
+  filter_count: 16
+  kernel_size: 9
+  kmer_length: 3
+serialization_version: 0
+training_hyperparameters:
+  learning_rate: 0.001
+  min_learning_rate: 1.0e-06
+  weight_decay: 1.0e-06
diff --git a/models/ThriftyHumV0.2-59.pth b/models/ThriftyHumV0.2-59.pth
diff --git a/models/ThriftyHumV0.2-59.yml b/models/ThriftyHumV0.2-59.yml
@@ -0,0 +1,16 @@
+encoder_class: KmerSequenceEncoder
+encoder_parameters:
+  kmer_length: 3
+  site_count: 500
+model_class: JoinedRSCNNModel
+model_hyperparameters:
+  dropout_prob: 0.3
+  embedding_dim: 7
+  filter_count: 19
+  kernel_size: 11
+  kmer_length: 3
+serialization_version: 0
+training_hyperparameters:
+  learning_rate: 0.001
+  min_learning_rate: 1.0e-06
+  weight_decay: 1.0e-06