catgrad 0.2.1

a categorical deep learning compiler
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

use catgrad::category::lang::*;
use catgrad::stdlib::*;
use catgrad::typecheck::*;

pub mod test_utils;
use test_utils::{get_forget_core_declarations, save_diagram_if_enabled};
pub mod test_models;
use test_models::LinearSigmoid;

#[test]
fn test_construct_linear_sigmoid() {
    let sigmoid = nn::Sigmoid.term().unwrap();
    println!("{sigmoid:?}");

    let term = LinearSigmoid.term().unwrap();
    println!("{term:?}");
}

#[test]
fn test_graph_sigmoid() {
    let term = nn::Sigmoid.term().unwrap().term;
    let term = open_hypergraphs::lax::var::forget::forget_monogamous(&term);
    save_diagram_if_enabled("test_graph_sigmoid.svg", &term);
}

#[test]
fn test_graph_linear_sigmoid() {
    let term = LinearSigmoid.term().unwrap().term;
    let term = open_hypergraphs::lax::var::forget::forget_monogamous(&term);
    save_diagram_if_enabled("test_graph_linear_sigmoid.svg", &term);
}

// Shapecheck the linear-sigmoid term.
// This should allow us to generate a diagram similar to the one in test_graph_linear_sigmoid(),
// but where objects are "symbolic shapes".
#[test]
fn test_check_linear_sigmoid() {
    run_check_test(LinearSigmoid.term(), "test_check_linear_sigmoid.svg").expect("valid");
}

#[test]
fn test_check_sigmoid() {
    run_check_test(nn::Sigmoid.term(), "test_check_sigmoid.svg").expect("valid");
}

#[test]
fn test_check_exp() {
    run_check_test(nn::Exp.term(), "test_check_exp.svg").expect("valid");
}

#[allow(clippy::result_large_err)]
pub fn run_check_test(
    term: Option<catgrad::category::lang::TypedTerm>,
    svg_filename: &str,
) -> Result<(), InterpreterError> {
    let TypedTerm {
        term, source_type, ..
    } = term.unwrap();

    let term = open_hypergraphs::lax::var::forget::forget_monogamous(&term);
    let env = get_forget_core_declarations();

    let result = check_with(&env, &Parameters::default(), term.clone(), source_type)?;
    println!("result: {result:?}");

    let typed_term = term
        .with_nodes(|_| result.into_iter().map(|e| format!("{e:?}")).collect())
        .unwrap();
    save_diagram_if_enabled(svg_filename, &typed_term);

    Ok(())
}

/*
#[test]
fn test_cyclic_definition_fails() {
    todo!()
}
*/