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

Commit

Permalink
enh: rename eddymotion._sklearn -> eddymotion.gpr
Browse files Browse the repository at this point in the history
* rename the module so that it is picked up by autodoc.
* also fixes minor errors and improves docs.
  • Loading branch information
oesteban committed Oct 29, 2024
1 parent 7e5f333 commit db5d51c
Show file tree
Hide file tree
Showing 6 changed files with 901 additions and 6 deletions.
4 changes: 2 additions & 2 deletions docs/notebooks/dmri_covariance.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
"import matplotlib.pyplot as plt\n",
"import numpy as np\n",
"\n",
"from eddymotion.model._sklearn import (\n",
"from eddymotion.model.gpr import (\n",
" compute_pairwise_angles,\n",
" exponential_covariance,\n",
" spherical_covariance,\n",
Expand Down Expand Up @@ -345,7 +345,7 @@
}
],
"source": [
"from eddymotion.model._sklearn import EddyMotionGPR, SphericalKriging\n",
"from eddymotion.model.gpr import EddyMotionGPR, SphericalKriging\n",
"\n",
"K = SphericalKriging(beta_a=PARAMETER_SPHERICAL_a, beta_l=PARAMETER_lambda)(X_real)\n",
"K -= K.min()\n",
Expand Down
4 changes: 2 additions & 2 deletions scripts/dwi_gp_estimation_error_analysis.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
import pandas as pd
from sklearn.model_selection import KFold, RepeatedKFold, cross_val_predict, cross_val_score

from eddymotion.model._sklearn import (
from eddymotion.model.gpr import (
EddyMotionGPR,
SphericalKriging,
)
Expand All @@ -63,7 +63,7 @@ def cross_validate(
Number of folds.
n_repeats : :obj:`int`
Number of times the cross-validator needs to be repeated.
gpr : obj:`~eddymotion.model._sklearn.EddyMotionGPR`
gpr : obj:`~eddymotion.model.gpr.EddyMotionGPR`
The eddymotion Gaussian process regressor object.
Returns
Expand Down
2 changes: 1 addition & 1 deletion scripts/dwi_gp_estimation_simulated_signal.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
import numpy as np
from dipy.core.sphere import Sphere

from eddymotion.model._sklearn import EddyMotionGPR, SphericalKriging
from eddymotion.model.gpr import EddyMotionGPR, SphericalKriging
from eddymotion.testing import simulations as testsims

SAMPLING_DIRECTIONS = 200
Expand Down
Loading

0 comments on commit db5d51c

Please sign in to comment.