diff --git a/fvgp/gp2Scale.py b/fvgp/gp2Scale.py index b51d06a..efbfd52 100755 --- a/fvgp/gp2Scale.py +++ b/fvgp/gp2Scale.py @@ -61,7 +61,6 @@ def ranges(N, nb): def compute_covariance(self, hyperparameters, client, batched=False): """computes the covariance matrix from the kernel on HPC in sparse format""" - NUM_RANGES = self.num_batches ranges = self.ranges(len(self.x_data), NUM_RANGES) # the chunk ranges, as (start, end) tuples ranges_ij = list( @@ -198,6 +197,7 @@ def __init__(self, input_space_dim, print("non zero elements: ", sp.nnz) print("Size in GBits: ", sp.data.nbytes / 1e9) print("Sparsity: ", sp.nnz / float(self.point_number) ** 2) + if self.point_number <= 5000: print("Here is an image:") plt.imshow(sp.toarray()) @@ -276,7 +276,6 @@ def compute_covariance(self, hyperparameters, variances, client): # del scatter_future ######### - def free_workers(self, futures, finished_futures): free_workers = set() remaining_futures = []