Skip to content

Commit b34af92

Browse files
committed
fix: shrink dataset and update config
1 parent 8a1db12 commit b34af92

File tree

4 files changed

+6
-1982
lines changed

4 files changed

+6
-1982
lines changed

models/rank/sign/config.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,14 @@ runner:
2727
infer_batch_size: 1024
2828
infer_reader_path: "sign_reader" # importlib format
2929
infer_load_path: "output_model_sign"
30-
infer_start_epoch: 7
31-
infer_end_epoch: 9
30+
infer_start_epoch: 9
31+
infer_end_epoch: 10
3232

3333

3434
hyper_parameters:
3535
pred_edges: 1 # !=0: use edges in dataset, 0: predict edges using L_0
3636
dim: 8 # dimension of entity and relation embeddings
37-
hidden_layer: 32 # neural hidden layer 64
37+
hidden_layer: 32 # neural hidden layer 32
3838
l0_para: [0.66, -0.1, 1.1] # l0 parameters, which are beta (temprature), zeta (interval_min) and gama (interval_max).
3939
l0_weight: 0.001 # weight of the l0 regularization term
4040
l2_weight: 0.001 # weight of the l2 regularization term

models/rank/sign/config_bigdata.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ runner:
3838
hyper_parameters:
3939
pred_edges: 1 # !=0: use edges in dataset, 0: predict edges using L_0
4040
dim: 8 # dimension of entity and relation embeddings
41-
hidden_layer: 32 # neural hidden layer 64
41+
hidden_layer: 32 # neural hidden layer 32
4242
l0_para: [0.66, -0.1, 1.1] # l0 parameters, which are beta (temprature), zeta (interval_min) and gama (interval_max).
4343
l0_weight: 0.001 # weight of the l0 regularization term
4444
l2_weight: 0.001 # weight of the l2 regularization term

0 commit comments

Comments
 (0)