Browse Source

[MNT] Add save spec in example

tags/v0.3.2
Gene 3 years ago
parent
commit
ae7ad79874
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      examples/example_pfs/main.py

+ 2
- 0
examples/example_pfs/main.py View File

@@ -154,6 +154,8 @@ class PFSDatasetWorkflow:
for idx in idx_list: for idx in idx_list:
train_x, train_y, test_x, test_y = pfs.get_idx_data(idx) train_x, train_y, test_x, test_y = pfs.get_idx_data(idx)
user_spec = specification.utils.generate_rkme_spec(X=test_x, gamma=0.1, cuda_idx=0) user_spec = specification.utils.generate_rkme_spec(X=test_x, gamma=0.1, cuda_idx=0)
user_spec_path = f"./user_spec/user_{idx}.json"
user_spec.save(user_spec_path)


user_info = BaseUserInfo( user_info = BaseUserInfo(
id=f"user_{idx}", semantic_spec=user_senmantic, stat_info={"RKMEStatSpecification": user_spec} id=f"user_{idx}", semantic_spec=user_senmantic, stat_info={"RKMEStatSpecification": user_spec}


Loading…
Cancel
Save