tract-onnx-opl 0.23.5
“Tiny, no-nonsense, self contained, TensorFlow and ONNX inference”
Contains an unbounded block loopRust test marker or harness
SHA-25647c06a347c6b7e0613cf0991f60ee45c1df24dec9708ecd2a85501f7a3a5fb1b
MaleculeH₂(DbMg)Md(Pa)
Evidence
1use std::convert::TryFrom;
2use std::convert::TryInto;
3use std::fmt::{self, Debug, Display};
4use std::iter;
5
6use tract_nnef::internal::*;
7
8use tract_ndarray::{
9 Array1, Array2, ArrayD, ArrayView1, ArrayView2, ArrayViewD, ArrayViewMut1, Axis, Ix1, Ix2,
10};
11
12use tract_num_traits::AsPrimitive;
13
14macro_rules! ensure {
15 ($cond: expr, $($rest: expr),* $(,)?) => {
16 if !$cond {
17 …
46:2… }
47 pub fn to_u8(&self) -> u8 {
48 unsafe { std::mem::transmute(*self) }
49 }
50}
51
52impl TryFrom<u8> for Cmp {
53 type Error = TractError;
54 fn try_from(value: u8) -> Result<Self, Self::Error> {
55 …
141:22… ode_id = self.trees.as_slice_unchecked::<u32>()[tree] as usize;
142 loop {
143 let node = self.get_unchecked(node_id);
144 match node {
145 TreeNode::Branch(ref b …
395:46… ty for tree ensemble: {:?}", input.shape());
396 }
397 }
398}
399
400#[cfg(test)]
401mod tests {
402 use super::*;
403 use tract_ndarray::prelude::*;
404
405 fn b(
⋯4 lines
572:10… [-0.147_204_7, -0.147_199_3, 0.281_595_9],
573 ])
574 }
575
576 #[test]
577 #[ignore]
578 fn test_tree_ensemble() {
579 let ensemble = generate_gbm_ensemble();
580 let input = generate_gbm_in …
No evidence locations were recorded for this file. Raw result