Skip to content

Commit

Permalink
Merge branch 'develop' of https://github.com/gizatechxyz/orion into d…
Browse files Browse the repository at this point in the history
…evelop
  • Loading branch information
raphaelDkhn committed Oct 5, 2023
2 parents 5f39efe + 2d4aa46 commit e4893c0
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 5 deletions.
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
# Tree Regressor

`TreeRegressorTrait` provides a trait definition for decision tree regression.
This trait offers functionalities to build a decision tree and predict target values based on input features.
`TreeRegressorTrait` provides a trait definition for decision tree regression. This trait offers functionalities to build a decision tree and predict target values based on input features.

```rust
use orion::operators::ml::TreeRegressorTrait;
Expand All @@ -21,4 +20,3 @@ Orion supports currently only fixed point data types for `TreeRegressorTrait`.
| --- | --- |
| [`tree.fit`](tree.fit.md) | Constructs a decision tree regressor based on the provided data and target values. |
| [`tree.predict`](tree.predict.md) | Given a set of features, predicts the target value using the constructed decision tree. |

Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# TreeRegressorTrait::predict
# tree.predict

```rust
```rust
fn predict(ref self: TreeNode<T>, features: Span<T>) -> T;
```

Expand Down

0 comments on commit e4893c0

Please sign in to comment.