我如何拯救keras模型從python在磚一個s3 bucket的筆記本嗎?

[manupmanoos]

我有一個訓練模型磚python的筆記本。我怎麼保存這個s3 bucket。

[庫瑪 · Accepted solution]

嗨@manupmanoos,

請檢查下麵的代碼如何加載保存模型從s3 bucket嗎

從keras進口boto3導入操作係統。模型導入load_model #憑證和創建客戶機aws_access_key_id = dbutils.secrets S3。get(範圍= " < scope-name >”,關鍵=“<鍵名>”)aws_secret_access_key = dbutils.secrets。get(範圍= " < scope-name >”,關鍵=“<鍵名>”)操作係統。環境[' AWS_ACCESS_KEY_ID '] = AWS_ACCESS_KEY_ID操作係統。環境[' AWS_SECRET_ACCESS_KEY '] = AWS_SECRET_ACCESS_KEY s3_client = boto3.client (s3) #指定s3 bucket和模型文件路徑s3_bucket =“< bucket名>”s3_prefix = " < bucket-prefix > " s3_key = " {} / model.h5 " .format (s3_prefix) #下載模型文件從s3 local_model_path = " / dbfs /模型/模型。s3_client h5”。download_file (s3_bucket s3_key local_model_path) #加載模型使用Keras loaded_model = load_model (local_model_path)