Grid search update (#158)

GeoStat-Framework · Aug 7, 2020 · a00297a · a00297a
1 parent b33e5d8
commit a00297a
Show file tree

Hide file tree

Showing 3 changed files with 227 additions and 72 deletions.
diff --git a/README.rst b/README.rst
@@ -12,7 +12,7 @@ PyKrige
 .. image:: https://coveralls.io/repos/github/GeoStat-Framework/PyKrige/badge.svg?branch=master
    :target: https://coveralls.io/github/GeoStat-Framework/PyKrige?branch=master
 .. image:: https://readthedocs.org/projects/pykrige/badge/?version=stable
-   :target: http://pykrige.readthedocs.io/en/latest/?badge=stable
+   :target: http://pykrige.readthedocs.io/en/stable/?badge=stable
    :alt: Documentation Status
 .. image:: https://img.shields.io/badge/code%20style-black-000000.svg
    :target: https://github.com/psf/black

diff --git a/examples/krige_cv.py b/examples/krige_cv.py
@@ -53,7 +53,7 @@
     # "weight": [True, False]
 }
 
-estimator = GridSearchCV(Krige(), param_dict3d, verbose=True)
+estimator = GridSearchCV(Krige(), param_dict3d, verbose=True, return_train_score=True)
 
 # dummy data
 X3 = np.random.randint(0, 400, size=(100, 3)).astype(float)