Skip to content

Commit

Permalink
Adding hpo example using optuna for hnsw_flat and cagra
Browse files Browse the repository at this point in the history
  • Loading branch information
dpadmanabhan03 committed Oct 8, 2024
1 parent 397e56e commit b1b2b65
Show file tree
Hide file tree
Showing 3 changed files with 488 additions and 17 deletions.
14 changes: 0 additions & 14 deletions notebooks/cuvs_hpo_example.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -11,20 +11,6 @@
"Note: This notebook has been tested on Sagemaker Studio with an instance type of ml.g5.12xlarge."
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "9d35868b-93ad-43b4-ae15-59e75aa89e3c",
"metadata": {
"tags": []
},
"outputs": [],
"source": [
"#Install Required Packages\n",
"%mamba install -c conda-forge -c nvidia -c rapidsai-nightly cuvs optuna -y\n",
"%pip install cupy"
]
},
{
"cell_type": "code",
"execution_count": 35,
Expand Down
Loading

0 comments on commit b1b2b65

Please sign in to comment.