Add loss transfer function to glvq

2021-05-04 20:56:16 +02:00
parent f402eea884
commit d644114090
2 changed files with 9 additions and 1 deletions
--- a/examples/glvq_spiral.py
+++ b/examples/glvq_spiral.py
@@ -35,6 +35,8 @@ if __name__ == "__main__":
        prototype_initializer=cinit.SSI(torch.Tensor(x_train),
                                        torch.Tensor(y_train),
                                        noise=1e-7),
+        transfer_function="sigmoid_beta",
+        transfer_beta=10.0,
        lr=0.01,
    )