From 831fd4a100c5bb25397aa42db0bad519648d5191 Mon Sep 17 00:00:00 2001 From: Gene Date: Fri, 31 Mar 2023 21:55:18 +0800 Subject: [PATCH] [ENH] Fill nan when generating RKMESpecification --- examples/examples1/example_rkme.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/examples1/example_rkme.py b/examples/examples1/example_rkme.py index ae0909e..69d7875 100644 --- a/examples/examples1/example_rkme.py +++ b/examples/examples1/example_rkme.py @@ -4,6 +4,8 @@ import learnware.specification as specification if __name__ == "__main__": data_X = np.random.randn(10000, 20) + for i in range(10): + data_X[i, i] = np.nan spec1 = specification.utils.generate_rkme_spec(X=data_X, gamma=0.1, cuda_idx=-1) spec2 = specification.rkme.RKMESpecification() spec1.generate_stat_spec_from_data(data_X)