DAGMA
dagma.nonlinear.DagmaMLP.h_func
Initializing search
    kevinsbello/dagma
    • Home
    • Getting Started
    • API Reference
    kevinsbello/dagma
      • Home
      • Getting Started
      • dagma
        • dagma.linear
          • DagmaLinear
            • _score
            • _h
            • _func
            • _adam_update
            • minimize
            • fit
        • dagma.locally_connected
          • LocallyConnected
            • reset_parameters
            • forward
            • extra_repr
        • dagma.nonlinear
          • DagmaMLP
            • forward
            • h_func
            • fc1_l1_reg
            • fc1_to_adj
          • DagmaNonlinear
            • log_mse_loss
            • minimize
            • fit
        • dagma.utils
          • set_random_seed
          • is_dag
          • simulate_dag
          • simulate_parameter
          • simulate_linear_sem
          • simulate_nonlinear_sem
          • count_accuracy

    dagma.nonlinear.DagmaMLP.h_func¶

    dagma.nonlinear.DagmaMLP.h_func(s: float = 1.0) → torch.Tensor¶

    Constrain 2-norm-squared of fc1 weights along m1 dim to be a DAG

    Parameters:¶
    s : float, optional¶

    Controls the domain of M-matrices, by default 1.0

    Returns:¶

    A scalar value of the log-det acyclicity function \(h(\Theta)\).

    Return type:¶

    torch.Tensor


    Last update: Jan 14, 2024
    Back to top
    Previous dagma.nonlinear.DagmaMLP.forward
    Next dagma.nonlinear.DagmaMLP.fc1_l1_reg
    © Copyright 2023, Kevin Bello.
    Created using Sphinx 7.1.2. and Sphinx-Immaterial