Function test_helper

ciphercore-base/src/ops/auc.rs:204–225

(y_true: Vec<i64>, y_pred: Vec<i64>)

    use crate::graphs::util::simple_context;

    fn test_helper(y_true: Vec<i64>, y_pred: Vec<i64>) -> Result<f64> {
        let array_t = array_type(vec![y_true.len() as u64], INT64);
        let c = simple_context(|g| {
            let y_true = g.input(array_t.clone())?;
            let y_pred = g.input(array_t)?;
            g.custom_op(
                CustomOperation::new(AucScore {
                    fp: FixedPrecisionConfig::default(),
                }),
                vec![y_true, y_pred],
            )
        })?;
        let mapped_c = run_instantiation_pass(c)?;
        let result = random_evaluate(
            mapped_c.get_context().get_main_graph()?,
            vec![
                Value::from_flattened_array(&y_true, INT64)?,
                Value::from_flattened_array(&y_pred, INT64)?,
            ],
        )?;
        Ok(result.to_i64(INT64)? as f64 / FixedPrecisionConfig::default().denominator_f64())
    }

    #[test]
    fn test_auc_simple_case() -> Result<()> {