From 25046e313ef449dd0dc75a54909fc01ef1ec42ef Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Radek=20Pu=C5=A1?= <pusradek@fit.cvut.cz>
Date: Thu, 16 Jan 2020 03:17:39 +0100
Subject: [PATCH] fix: amount prediction - not-repetitive transaction has not
 been filtered out

---
 .../Controllers/containers/ForecastTransactionContainer.cs    | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/Core/Core/Controllers/containers/ForecastTransactionContainer.cs b/Core/Core/Controllers/containers/ForecastTransactionContainer.cs
index b015cf6..10be7d2 100644
--- a/Core/Core/Controllers/containers/ForecastTransactionContainer.cs
+++ b/Core/Core/Controllers/containers/ForecastTransactionContainer.cs
@@ -44,7 +44,7 @@ namespace Core.Controllers.containers
                     });
             }
 
-            FillAmountPredictionLists(amountPrediction);
+            FillAmountPredictionLists(amountPrediction.Where(x => x.Item3).ToList());
 
             this.neuralPrediction = neuralPrediction;
             this.neuralPrediction.ForEach(c => c.sizeType = GetTransactionTypesName(Math.Abs(c.amount)));
@@ -65,7 +65,6 @@ namespace Core.Controllers.containers
             amountPredictionSizeTypes = MakeSizeTypesList(amountPrediction);
 
             amountPredictionList = amountPrediction
-                .Where(x => x.Item3)
                 .Select(x => new TransactionContainer()
                 {
                     amount = x.Item1,
@@ -73,7 +72,6 @@ namespace Core.Controllers.containers
                 }).ToList();
 
             amountPredictionKinds = amountPrediction
-                .Where(t => t.Item3)
                 .GroupBy(t => t.Item2)
                 .Select(g => new TransactionContainer()
                 {
-- 
GitLab