diff --git a/docs/framework/operators/machine-learning/tree-regressor/README.md b/docs/framework/operators/machine-learning/tree-regressor/README.md index e2e69b355..7df2112c4 100644 --- a/docs/framework/operators/machine-learning/tree-regressor/README.md +++ b/docs/framework/operators/machine-learning/tree-regressor/README.md @@ -20,3 +20,4 @@ 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. | + diff --git a/docs/framework/operators/machine-learning/tree-regressor/tree.predict.md b/docs/framework/operators/machine-learning/tree-regressor/tree.predict.md index 0d14f381a..28d4a027c 100644 --- a/docs/framework/operators/machine-learning/tree-regressor/tree.predict.md +++ b/docs/framework/operators/machine-learning/tree-regressor/tree.predict.md @@ -1,6 +1,6 @@ -# tree.predict +# TreeRegressorTrait::predict -```rust +```rust fn predict(ref self: TreeNode, features: Span) -> T; ``` diff --git a/docs/framework/operators/tensor/tensor.sign.md b/docs/framework/operators/tensor/tensor.sign.md index ef35104f5..8acb8c730 100644 --- a/docs/framework/operators/tensor/tensor.sign.md +++ b/docs/framework/operators/tensor/tensor.sign.md @@ -5,6 +5,7 @@ ``` Calculates the sign of the given input tensor element-wise. +If input > 0, output 1. if input < 0, output -1. if input == 0, output 0. ## Args diff --git a/src/operators/tensor/core.cairo b/src/operators/tensor/core.cairo index 0d658ed53..eb52c6c1c 100644 --- a/src/operators/tensor/core.cairo +++ b/src/operators/tensor/core.cairo @@ -77,6 +77,7 @@ impl TensorSerde, impl TDrop: Drop> of Serde { /// # tensor.new @@ -2646,6 +2647,7 @@ trait TensorTrait { /// ``` /// /// Calculates the sign of the given input tensor element-wise. + /// If input > 0, output 1. if input < 0, output -1. if input == 0, output 0. /// /// ## Args ///