def test_log_feature_importance_with_limit(limit: int) -> None:
with mlflow.start_run() as run:
default_path = _get_default_args(lg.log_feature_importance)["path"]
lg.log_feature_importance(["a", "b", "c"], [1, 2, 3],
"gain",
limit=limit)
assert_file_exists_in_artifacts(run, default_path)
def test_log_confusion_matrix() -> None:
with mlflow.start_run() as run:
default_path = _get_default_args(lg.log_confusion_matrix)["path"]
lg.log_confusion_matrix([[1, 2], [3, 4]])
assert_file_exists_in_artifacts(run, default_path)
with mlflow.start_run() as run:
path = "cm.png"
lg.log_confusion_matrix([[1, 2], [3, 4]], path)
assert_file_exists_in_artifacts(run, path)
def test_log_pr_curve() -> None:
default_path = _get_default_args(lg.log_pr_curve)["path"]
with mlflow.start_run() as run:
lg.log_pr_curve([1, 0], [1, 0])
assert_file_exists_in_artifacts(run, default_path)
with mlflow.start_run() as run:
lg.log_pr_curve([1, 0], [1, 0], 0.5)
assert_file_exists_in_artifacts(run, default_path)
with mlflow.start_run() as run:
path = "pr.png"
lg.log_pr_curve([1, 0], [1, 0], path=path)
assert_file_exists_in_artifacts(run, path)
def test_log_roc_curve():
default_path = _get_default_args(lg.log_roc_curve)["path"]
with mlflow.start_run() as run:
lg.log_roc_curve([0, 1], [0, 1])
assert_file_exists_in_artifacts(run, default_path)
with mlflow.start_run() as run:
lg.log_roc_curve([0, 1], [0, 1], 0.5)
assert_file_exists_in_artifacts(run, default_path)
with mlflow.start_run() as run:
path = "roc.png"
lg.log_roc_curve([0, 1], [0, 1], path=path)
assert_file_exists_in_artifacts(run, path)
def test_log_feature_importance():
default_path = _get_default_args(lg.log_feature_importance)["path"]
with mlflow.start_run() as run:
lg.log_feature_importance(["a", "b", "c"], [1, 2, 3], "gain")
assert_file_exists_in_artifacts(run, default_path)
with mlflow.start_run() as run:
lg.log_feature_importance(["a", "b", "c"], [1, 2, 3], "gain", normalize=True)
assert_file_exists_in_artifacts(run, default_path)
with mlflow.start_run() as run:
path = "feat_imp.png"
lg.log_feature_importance(["a", "b", "c"], [1, 2, 3], "gain", path=path)
assert_file_exists_in_artifacts(run, path)