Skip to content

Commit

Permalink
update doc
Browse files Browse the repository at this point in the history
  • Loading branch information
raphaelDkhn committed Oct 6, 2023
1 parent 1231a26 commit 031c25e
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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. |

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

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

Expand Down
1 change: 1 addition & 0 deletions docs/framework/operators/tensor/tensor.sign.md
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
2 changes: 2 additions & 0 deletions src/operators/tensor/core.cairo
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,7 @@ impl TensorSerde<T, impl TSerde: Serde<T>, impl TDrop: Drop<T>> of Serde<Tensor<
/// squeeze - Removes dimensions of size 1 from the shape of a tensor.
/// unsqueeze - Inserts single-dimensional entries to the shape of an input tensor.
/// sign - Calculates the sign of the given input tensor element-wise.
/// clip - Clip operator limits the given input within an interval.
///
trait TensorTrait<T> {
/// # tensor.new
Expand Down Expand Up @@ -2646,6 +2647,7 @@ trait TensorTrait<T> {
/// ```
///
/// 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
///
Expand Down

0 comments on commit 031c25e

Please sign in to comment.