From 63e9c38beb5083e620bdbe9ef1cf5fc75cba8bc4 Mon Sep 17 00:00:00 2001
From: David Luevano Alvarado <55825613+luevano@users.noreply.github.com>
Date: Thu, 26 Mar 2020 15:51:27 -0700
Subject: Fix kernels
---
ml_exp/__init__.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py
index b40957baf..e1d3bcd6f 100644
--- a/ml_exp/__init__.py
+++ b/ml_exp/__init__.py
@@ -25,7 +25,7 @@ from ml_exp.representations import coulomb_matrix, lennard_jones_matrix,\
get_helping_data, adjacency_matrix, epsilon_index, check_bond, bag_of_bonds
from ml_exp.readdb import qm7db, qm9db
from ml_exp.data import NUCLEAR_CHARGE, POSSIBLE_BONDS
-from ml_exp.kernels import laplauss_kernel
+from ml_exp.kernels import gaussian_kernel, laplauss_kernel, wasserstein_kernel
from ml_exp.krr import krr, multi_krr
__all__ = ['Compound',
@@ -38,7 +38,9 @@ __all__ = ['Compound',
'bag_of_bonds',
'qm7db',
'qm9db',
- 'laplauss_kernel',
+ 'gaussian_kernel',
+ 'laplacian_kernel',
+ 'wasserstein_kernel',
'krr',
'multi_krr',
'NUCLEAR_CHARGE',
--
cgit v1.2.3-70-g09d2