通过 GitHub 从 mlruns 文件夹加载 ML-Model

问题描述 投票:0回答:0

在我的本地机器上,我已经调整并记录了 ML 模型的超参数,并使用 mlruns 文件夹创建了一个公共 GitHub 存储库。

因为我还想在 Google Collab 环境中使用该模型,所以我想弄清楚如何从我的公共 GitHub 存储库加载该模型。

本地,与

# Import and training of the best-tuned model from the MLflow registry
model_name = "model-XYZ"
model_version = 1

model = mlflow.sklearn.load_model(f"models:/{model_name}/{model_version}")
_ = model.fit(X_train, y_train)

这不是问题。但是对于 GitHub,我不知道路径应该是什么样子?以前的尝试,例如与

github_repo = "myRepo"
run_id = "myrunID"
model_name = "myModelName"

model_uri = f"https://raw.githubusercontent.com/{github_repo}/blob/main/mlruns/0/{run_id}/artifacts/{model_name}"

model = mlflow.sklearn.load_model(model_uri)

没用。

编辑:现在我通过简单地将整个 repo 克隆到我的谷歌协作并导航到根目录就像在我的本地项目中一样帮助自己。但是,现在它无法识别(未找到)我尝试提供的模型名称,即使该文件夹确实存在并且也在本地工作

感谢您的帮助!

scikit-learn ml mlflow
最新问题
© www.soinside.com 2019 - 2025. All rights reserved.