From 6aca56da6a9306e1ee594836a21f61558b5cb9ce Mon Sep 17 00:00:00 2001 From: Janis Klaise Date: Thu, 12 Oct 2023 15:35:59 +0100 Subject: [PATCH] Fix test models to output logits and work with default loss functions --- alibi/explainers/tests/test_simiarlity/conftest.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/alibi/explainers/tests/test_simiarlity/conftest.py b/alibi/explainers/tests/test_simiarlity/conftest.py index 4eb407e3f..863501580 100644 --- a/alibi/explainers/tests/test_simiarlity/conftest.py +++ b/alibi/explainers/tests/test_simiarlity/conftest.py @@ -151,7 +151,6 @@ def tf_linear_model(input_shape, output_shape): tf.keras.layers.InputLayer(input_shape=input_shape), tf.keras.layers.Flatten(), tf.keras.layers.Dense(output_shape), - tf.keras.layers.Softmax() ]) @@ -167,7 +166,6 @@ def __init__(self, input_shape, output_shape): self.linear_stack = nn.Sequential( nn.Flatten(start_dim=1), nn.Linear(input_shape, output_shape), - nn.Softmax() ) def forward(self, x):