Skip to content
This repository was archived by the owner on Dec 8, 2024. It is now read-only.

Commit d1d6d23

Browse files
authored
Fixes problem with gaussian kernel and QMLearn (#118)
1 parent 834541f commit d1d6d23

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

qml/kernels/fkernels.f90

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -541,7 +541,7 @@ subroutine fgaussian_kernel_symmetric(x, n, k, sigma)
541541

542542
!$OMP PARALLEL DO PRIVATE(temp, val) SCHEDULE(dynamic)
543543
do i = 1, n
544-
do j = 1, n
544+
do j = i, n
545545
temp = x(:,j) - x(:,i)
546546
val = exp(inv_sigma * dot_product(temp,temp))
547547
k(j,i) = val

qml/qmlearn/kernels.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -358,9 +358,9 @@ def generate(self, X, Y=None, representation_type='molecular'):
358358
def _generate_molecular(self, X, Y=None):
359359
if Y is None or X is Y:
360360
# Do symmetric matrix
361-
return gaussian_kernel_symmetric(X, self.sigma)
361+
return gaussian_kernel_symmetric(np.array(X), self.sigma)
362362
else:
363-
return gaussian_kernel(X, Y, self.sigma)
363+
return gaussian_kernel(np.array(X), np.array(Y), self.sigma)
364364

365365
def _generate_atomic(self, X, Y=None):
366366

0 commit comments

Comments
 (0)