diff --git a/lightning_logs/pre-trained/best_result.txt b/lightning_logs/pre-trained/best_result.txt new file mode 100644 index 0000000..8f0a6be --- /dev/null +++ b/lightning_logs/pre-trained/best_result.txt @@ -0,0 +1 @@ +{'auc': 0.9483146071434021, 'aupr': 0.9530767202377319} \ No newline at end of file diff --git a/lightning_logs/pre-trained/checkpoints/pre-trained.z01 b/lightning_logs/pre-trained/checkpoints/pre-trained.z01 new file mode 100644 index 0000000..1f95651 Binary files /dev/null and b/lightning_logs/pre-trained/checkpoints/pre-trained.z01 differ diff --git a/lightning_logs/pre-trained/checkpoints/pre-trained.z02 b/lightning_logs/pre-trained/checkpoints/pre-trained.z02 new file mode 100644 index 0000000..fdc9611 Binary files /dev/null and b/lightning_logs/pre-trained/checkpoints/pre-trained.z02 differ diff --git a/lightning_logs/pre-trained/checkpoints/pre-trained.zip b/lightning_logs/pre-trained/checkpoints/pre-trained.zip new file mode 100644 index 0000000..f13f4f4 Binary files /dev/null and b/lightning_logs/pre-trained/checkpoints/pre-trained.zip differ diff --git a/lightning_logs/pre-trained/hparams.yaml b/lightning_logs/pre-trained/hparams.yaml new file mode 100644 index 0000000..48635aa --- /dev/null +++ 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