Esempio n. 1
0
 def test_plot_sufficient_summary_02(self):
     nll = NonlinearLevelSet(n_layers=2, active_dim=2, lr=0.02, epochs=1)
     nll.train(inputs=inputs_torch, gradients=grad_torch, interactive=False)
     with self.assertRaises(ValueError):
         nll.plot_sufficient_summary(inputs=inputs_torch, outputs=lift)
Esempio n. 2
0
 def test_plot_sufficient_summary_01(self):
     nll = NonlinearLevelSet(n_layers=2, active_dim=1, lr=0.02, epochs=1)
     nll.train(inputs=inputs_torch, gradients=grad_torch, interactive=False)
     with assert_plot_figures_added():
         nll.plot_sufficient_summary(inputs=inputs_torch, outputs=lift)