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

Commit

Permalink
STYLE: Lint DWI GP estimation notebook
Browse files Browse the repository at this point in the history
Lint DWI GP estimation notebook.

Take advantage of the commit to remove the unused random number
generator.
  • Loading branch information
jhlegarreta committed Nov 9, 2024
1 parent bca7e8a commit cbff151
Showing 1 changed file with 26 additions and 19 deletions.
45 changes: 26 additions & 19 deletions docs/notebooks/dwi_gp_estimation.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -29,23 +29,9 @@
"from dipy.core.gradients import get_bval_indices\n",
"from dipy.io import read_bvals_bvecs\n",
"from dipy.segment.mask import median_otsu\n",
"\n",
"from scipy.ndimage import binary_dilation\n",
"from skimage.morphology import ball\n",
"\n",
"from nireports.reportlets.modality.dwi import plot_dwi"
]
},
{
"cell_type": "code",
"execution_count": 2,
"id": "69a3bc6b4fbe7036",
"metadata": {},
"outputs": [],
"source": [
"seed = 1234\n",
"rng = np.random.default_rng(seed)\n",
"\n",
"name = \"sherbrooke_3shell\"\n",
"\n",
"dwi_fname, bval_fname, bvec_fname = dpd.get_fnames(name=name)\n",
Expand Down Expand Up @@ -131,8 +117,15 @@
}
],
"source": [
"from nireports.reportlets.modality.dwi import plot_dwi\n",
"\n",
"affine = nib.load(dwi_fname).affine\n",
"plot_dwi(shell_data[..., dwi_vol_idx], affine, gradient=np.concatenate((np.squeeze(X_test), [1000])), output_file=\"sherbrooke_3shell_b1k_data.svg\")"
"plot_dwi(\n",
" shell_data[..., dwi_vol_idx],\n",
" affine,\n",
" gradient=np.concatenate((np.squeeze(X_test), [1000])),\n",
" output_file=\"sherbrooke_3shell_b1k_data.svg\",\n",
")"
]
},
{
Expand All @@ -159,7 +152,7 @@
"alpha = 0.1\n",
"disp = True\n",
"optimizer = None\n",
"gpr = EddyMotionGPR(kernel=kernel, alpha=alpha, disp=disp, optimizer=optimizer)\n"
"gpr = EddyMotionGPR(kernel=kernel, alpha=alpha, disp=disp, optimizer=optimizer)"
]
},
{
Expand Down Expand Up @@ -221,7 +214,14 @@
"output_type": "display_data"
}
],
"source": "plot_dwi(dwi_sim, affine, gradient=np.concatenate((np.squeeze(X_test), [1000])), output_file=\"sherbrooke_3shell_b1k_gp_noopt_pred.svg\")"
"source": [
"plot_dwi(\n",
" dwi_sim,\n",
" affine,\n",
" gradient=np.concatenate((np.squeeze(X_test), [1000])),\n",
" output_file=\"sherbrooke_3shell_b1k_gp_noopt_pred.svg\",\n",
")"
]
},
{
"cell_type": "code",
Expand Down Expand Up @@ -449,8 +449,15 @@
],
"source": [
"plot_dwi(dwi_sim, affine, gradient=np.concatenate((np.squeeze(X_test), [1000])))\n",
"plot_dwi(shell_data[..., dwi_vol_idx], affine, gradient=np.concatenate((np.squeeze(X_test), [1000])))\n",
"plot_dwi(dwi_sim2, affine, gradient=np.concatenate((np.squeeze(X_test), [1000])), output_file=\"sherbrooke_3shell_b1k_gp_opt_pred.svg\")"
"plot_dwi(\n",
" shell_data[..., dwi_vol_idx], affine, gradient=np.concatenate((np.squeeze(X_test), [1000]))\n",
")\n",
"plot_dwi(\n",
" dwi_sim2,\n",
" affine,\n",
" gradient=np.concatenate((np.squeeze(X_test), [1000])),\n",
" output_file=\"sherbrooke_3shell_b1k_gp_opt_pred.svg\",\n",
")"
]
}
],
Expand Down

0 comments on commit cbff151

Please sign in to comment.