Skip to content

ml.py: Cleanup #2449

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
schwehr merged 1 commit into from
Jan 7, 2026
Merged

ml.py: Cleanup #2449

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 6 additions & 5 deletions geemap/ml.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,9 @@
import pandas as pd


def tree_to_string(estimator, feature_names, labels=None, output_mode="INFER"):
def tree_to_string(
estimator, feature_names, labels=None, output_mode: str = "INFER"
) -> str:
"""Convert a sklearn decision tree object to a string format that EE can interpret

Args:
Expand All @@ -24,15 +26,14 @@ def tree_to_string(estimator, feature_names, labels=None, output_mode="INFER"):
(i.e., bands) used to create the model
labels (Iterable[numeric]): List of class labels to map outputs to, must be
numeric values. If None, then raw outputs will be used. default = None
output_mode (str): the output mode of the estimator. Options are "INFER",
output_mode: the output mode of the estimator. Options are "INFER",
"CLASSIFIATION", or "REGRESSION" (capitalization does not matter). default =
"INFER"

Returns:
tree_str (str): string representation of decision tree estimator

Raises:

RuntimeError: raises run time error when function cannot determine if the
estimator is for regression or classification problem.
"""
Expand Down Expand Up @@ -396,7 +397,7 @@ def export_trees_to_fc(trees, asset_id, description="geemap_rf_export"):
task.start()


def trees_to_csv(trees: list[str], out_csv: str):
def trees_to_csv(trees: list[str], out_csv: str) -> None:
"""Save a list of strings (an ensemble of decision trees) to a CSV file.

Args:
Expand All @@ -408,7 +409,7 @@ def trees_to_csv(trees: list[str], out_csv: str):
f.writelines([tree.replace("\n", "#") + "\n" for tree in trees])


def csv_to_classifier(in_csv: str) -> ee.Classifier:
def csv_to_classifier(in_csv: str) -> ee.Classifier | None:
"""Returns an ee.Classifier from a CSV file.

The file must contain a list of strings (an ensemble of decision trees).
Expand Down