add pre-trained model

farkguidao · Dec 31, 2021 · 1cf7797 · 1cf7797
1 parent a8a6be0
commit 1cf7797
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 0 deletions.
diff --git a/lightning_logs/pre-trained/best_result.txt b/lightning_logs/pre-trained/best_result.txt
@@ -0,0 +1 @@
+{'auc': 0.9483146071434021, 'aupr': 0.9530767202377319}
diff --git a/lightning_logs/pre-trained/checkpoints/pre-trained.z01 b/lightning_logs/pre-trained/checkpoints/pre-trained.z01
diff --git a/lightning_logs/pre-trained/checkpoints/pre-trained.z02 b/lightning_logs/pre-trained/checkpoints/pre-trained.z02
diff --git a/lightning_logs/pre-trained/checkpoints/pre-trained.zip b/lightning_logs/pre-trained/checkpoints/pre-trained.zip
diff --git a/lightning_logs/pre-trained/hparams.yaml b/lightning_logs/pre-trained/hparams.yaml
@@ -0,0 +1,15 @@
+K: 3
+N: 21257
+dropout: 0.1
+em_dim: 32
+feature_dim_list:
+- 196
+- 128
+hidden_dim: 64
+lam: 0.5
+learning_rate: 0.01
+node_num_list:
+- 19122
+- 2135
+pos_weight: 1.0
+weight_decay: 0.0001
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		{'auc': 0.9483146071434021, 'aupr': 0.9530767202377319}