in credit-card-fraud-detection/neptune_ml_utils.py [0:0]
def get_performance_metrics(training_job_name: str, download_location: str = './model-artifacts'):
training_job_s3_output = get_modeltraining_job_output_location(training_job_name)
if not training_job_s3_output:
return
download_location = os.path.join(download_location, training_job_name)
os.makedirs(download_location, exist_ok=True)
# download embeddings and mapping info
S3Downloader.download(os.path.join(training_job_s3_output, "eval_metrics_info.json"),
download_location)
with open(os.path.join(download_location, "eval_metrics_info.json")) as f:
metrics = json.load(f)
return metrics