Skip to content

Commit

Permalink
refactoring and cleaning
Browse files Browse the repository at this point in the history
  • Loading branch information
Remi Tschupp committed Jul 1, 2024
1 parent d0e2d70 commit f9aa906
Showing 1 changed file with 1 addition and 29 deletions.
30 changes: 1 addition & 29 deletions titanic/titanic_models.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@
},
{
"cell_type": "code",
"execution_count": 3,
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
Expand Down Expand Up @@ -449,34 +449,6 @@
"predicted_data[\"Survived\"] = predictions\n",
"print(predicted_data)"
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {},
"outputs": [
{
"ename": "MlflowException",
"evalue": "Registered Model with name=tracking-quickstart not found",
"output_type": "error",
"traceback": [
"\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
"\u001b[1;31mMlflowException\u001b[0m Traceback (most recent call last)",
"Cell \u001b[1;32mIn[9], line 3\u001b[0m\n\u001b[0;32m 1\u001b[0m get_ipython()\u001b[38;5;241m.\u001b[39msystem(\u001b[38;5;124m'\u001b[39m\u001b[38;5;124m cd ..\u001b[39m\u001b[38;5;124m'\u001b[39m)\n\u001b[0;32m 2\u001b[0m client \u001b[38;5;241m=\u001b[39m MlflowClient()\n\u001b[1;32m----> 3\u001b[0m \u001b[38;5;28mprint\u001b[39m(\u001b[43mclient\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mget_registered_model\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mtracking-quickstart\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m)\u001b[49m)\n\u001b[0;32m 4\u001b[0m \u001b[38;5;66;03m# Delete a registered model along with all its versions\u001b[39;00m\n\u001b[0;32m 5\u001b[0m client\u001b[38;5;241m.\u001b[39mdelete_registered_model(name\u001b[38;5;241m=\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mtracking-quickstart\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n",
"File \u001b[1;32mc:\\Users\\RT277831\\Documents\\Projets\\Dauphine\\ML_OPS\\venv\\Lib\\site-packages\\mlflow\\tracking\\client.py:3394\u001b[0m, in \u001b[0;36mMlflowClient.get_registered_model\u001b[1;34m(self, name)\u001b[0m\n\u001b[0;32m 3353\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mget_registered_model\u001b[39m(\u001b[38;5;28mself\u001b[39m, name: \u001b[38;5;28mstr\u001b[39m) \u001b[38;5;241m-\u001b[39m\u001b[38;5;241m>\u001b[39m RegisteredModel:\n\u001b[0;32m 3354\u001b[0m \u001b[38;5;250m \u001b[39m\u001b[38;5;124;03m\"\"\"Get a registered model.\u001b[39;00m\n\u001b[0;32m 3355\u001b[0m \n\u001b[0;32m 3356\u001b[0m \u001b[38;5;124;03m Args:\u001b[39;00m\n\u001b[1;32m (...)\u001b[0m\n\u001b[0;32m 3392\u001b[0m \u001b[38;5;124;03m description: This sentiment analysis model classifies the tone-happy, sad, angry.\u001b[39;00m\n\u001b[0;32m 3393\u001b[0m \u001b[38;5;124;03m \"\"\"\u001b[39;00m\n\u001b[1;32m-> 3394\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_get_registry_client\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mget_registered_model\u001b[49m\u001b[43m(\u001b[49m\u001b[43mname\u001b[49m\u001b[43m)\u001b[49m\n",
"File \u001b[1;32mc:\\Users\\RT277831\\Documents\\Projets\\Dauphine\\ML_OPS\\venv\\Lib\\site-packages\\mlflow\\tracking\\_model_registry\\client.py:137\u001b[0m, in \u001b[0;36mModelRegistryClient.get_registered_model\u001b[1;34m(self, name)\u001b[0m\n\u001b[0;32m 129\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mget_registered_model\u001b[39m(\u001b[38;5;28mself\u001b[39m, name):\n\u001b[0;32m 130\u001b[0m \u001b[38;5;250m \u001b[39m\u001b[38;5;124;03m\"\"\"\u001b[39;00m\n\u001b[0;32m 131\u001b[0m \u001b[38;5;124;03m Args:\u001b[39;00m\n\u001b[0;32m 132\u001b[0m \u001b[38;5;124;03m name: Name of the registered model to get.\u001b[39;00m\n\u001b[1;32m (...)\u001b[0m\n\u001b[0;32m 135\u001b[0m \u001b[38;5;124;03m A single :py:class:`mlflow.entities.model_registry.RegisteredModel` object.\u001b[39;00m\n\u001b[0;32m 136\u001b[0m \u001b[38;5;124;03m \"\"\"\u001b[39;00m\n\u001b[1;32m--> 137\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mstore\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mget_registered_model\u001b[49m\u001b[43m(\u001b[49m\u001b[43mname\u001b[49m\u001b[43m)\u001b[49m\n",
"File \u001b[1;32mc:\\Users\\RT277831\\Documents\\Projets\\Dauphine\\ML_OPS\\venv\\Lib\\site-packages\\mlflow\\store\\model_registry\\file_store.py:404\u001b[0m, in \u001b[0;36mFileStore.get_registered_model\u001b[1;34m(self, name)\u001b[0m\n\u001b[0;32m 402\u001b[0m model_path \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_get_registered_model_path(name)\n\u001b[0;32m 403\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m exists(model_path):\n\u001b[1;32m--> 404\u001b[0m \u001b[38;5;28;01mraise\u001b[39;00m MlflowException(\n\u001b[0;32m 405\u001b[0m \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mRegistered Model with name=\u001b[39m\u001b[38;5;132;01m{\u001b[39;00mname\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m not found\u001b[39m\u001b[38;5;124m\"\u001b[39m,\n\u001b[0;32m 406\u001b[0m RESOURCE_DOES_NOT_EXIST,\n\u001b[0;32m 407\u001b[0m )\n\u001b[0;32m 408\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m_get_registered_model_from_path(model_path)\n",
"\u001b[1;31mMlflowException\u001b[0m: Registered Model with name=tracking-quickstart not found"
]
}
],
"source": [
"registry_uri = \n",
"client = MlflowClient()\n",
"print(client.get_registered_model(\"tracking-quickstart\"))\n",
"# Delete a registered model along with all its versions\n",
"client.delete_registered_model(name=\"tracking-quickstart\")"
]
}
],
"metadata": {
Expand Down

0 comments on commit f9aa906

Please sign in to comment.