change position of softmax in get_predictions

Author: maxkeller321

modAL/dropout.py

@@ -315,16 +315,15 @@ def get_predictions(classifier: BaseEstimator, X: modALinput, dropout_layer_inde
             #In comparison to: predict(), predict_proba() the infer() 
             # does not change train/eval mode of other layers 
             logits = classifier.estimator.infer(samples)
-
             prediction = logits_adaptor(logits, samples)
-            mask = ~prediction.isnan()
-            prediction[mask] = prediction[mask].unsqueeze(0).softmax(1)
 
             if probas is None: probas = torch.empty((number_of_samples, prediction.shape[-1]))
-
             probas[range(sample_per_forward_pass*index, sample_per_forward_pass*(index+1)), :] = prediction
 
-        probas = to_numpy(prediction)
+
+        mask = ~probas.isnan()
+        probas[mask] = probas[mask].unsqueeze(0).softmax(1)
+        probas = to_numpy(probas)
         predictions.append(probas)
 
     # set dropout layers to eval