Skip to content
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.

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

Add ML processes for 2.1.0 #418

Closed
wants to merge 4 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
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
3 changes: 2 additions & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
- `fit_regr_random_forest`
- `flatten_dimensions`
- `load_ml_model`
- `predict_random_forest`
- `predict_ml_model`
- `predict_ml_model_probabilities`
- `save_ml_model`
- `unflatten_dimension`
- `vector_buffer`
Expand Down
2 changes: 1 addition & 1 deletion proposals/load_ml_model.json
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
}
],
"returns": {
"description": "A machine learning model to be used with machine learning processes such as ``predict_random_forest()``.",
"description": "A machine learning model to be used with machine learning processes such as ``predict_ml_model()`` or ``predict_ml_model_probabilities()``.",
"schema": {
"type": "object",
"subtype": "ml-model"
Expand Down
2 changes: 1 addition & 1 deletion proposals/predict_curve.json
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
{
"id": "predict_curve",
"summary": "Predict values",
"summary": "Predict values using a model function",
"description": "Predict values using a model function and pre-computed parameters. The process is primarily intended to compute values for new labels, but it can also fill gaps where existing labels contain no-data (`null`) values.",
"categories": [
"cubes",
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
{
"id": "predict_random_forest",
"summary": "Predict values based on a Random Forest model",
"description": "Applies a Random Forest machine learning model to an array and predict a value for it.",
"id": "predict_ml_model",
"summary": "Predict values values using a ML model",
"description": "Applies a machine learning model to an array and predicts a value/class for it.",
"categories": [
"machine learning",
"reducer"
Expand All @@ -23,7 +23,7 @@
},
{
"name": "model",
"description": "A model object that can be trained with the processes ``fit_regr_random_forest()`` (regression) and ``fit_class_random_forest()`` (classification).",
"description": "A ML model that can be trained with one of the ML processes such as ``fit_class_random_forest()``.",
"schema": {
"type": "object",
"subtype": "ml-model"
Expand Down
45 changes: 45 additions & 0 deletions proposals/predict_ml_model_probabilities.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
{
"id": "predict_ml_model_probabilities",
"summary": "Predict class probabilities using a ML model",
"description": "Applies a machine learning model to an array and predicts (class) probabilities for them.",
"categories": [
"machine learning",
"reducer"
],
"experimental": true,
"parameters": [
{
"name": "data",
"description": "An array of numbers.",
"schema": {
"type": "array",
"items": {
"type": [
"number",
"null"
]
}
}
},
{
"name": "model",
"description": "A ML model that can be trained with one of the ML processes such as ``fit_regr_random_forest()``.",
"schema": {
"type": "object",
"subtype": "ml-model"
}
}
],
"returns": {
"description": "The predicted (class) probabilities. Returns `null` if any of the given values in the array is a no-data value.",
"schema": {
"type": "array",
"items": {
"type": [
"number",
"null"
]
}
}
}
}