diff --git a/examples/example_pfs/main.py b/examples/example_pfs/main.py index fd943c9..c6708db 100644 --- a/examples/example_pfs/main.py +++ b/examples/example_pfs/main.py @@ -154,6 +154,8 @@ class PFSDatasetWorkflow: for idx in idx_list: 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_path = f"./user_spec/user_{idx}.json" + user_spec.save(user_spec_path) user_info = BaseUserInfo( id=f"user_{idx}", semantic_spec=user_senmantic, stat_info={"RKMEStatSpecification": user_spec}