如何重命名通过MLFlow记录的工件?

时间:2019-12-13 18:21:02

标签: python json file temporary-files mflow

我正在使用一个临时文件,然后将我的数据转储到json,然后将该数据作为工件记录到GCP中。临时文件名称是随机生成的,因此有没有办法将其重命名为我选择的自定义名称,以便具有正确命名约定的工件被MLFlow记录?

代码示例:

 with mlflow.start_run():
        with tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False) as f:
            original_path = f.name
            new_filename = 'metadata_' + pipeline_run_id
            json.dump(user_args, f)

            f.name = new_filename

            f.seek(0)
            mlflow.log_artifact(f.name)

            os.unlink(original_path)
            assert not os.path.exists(original_path)

0 个答案:

没有答案