From 72d8d396f3ad381f88c17fbf7b80e99506987e3e Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Wed, 17 Feb 2021 17:04:28 +0100
Subject: [PATCH] Added NA to default value in NumericColumnModule
---
torch_modules/src/NumericColumnModule.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/torch_modules/src/NumericColumnModule.cpp b/torch_modules/src/NumericColumnModule.cpp
index 29138d5..4899d59 100644
--- a/torch_modules/src/NumericColumnModule.cpp
+++ b/torch_modules/src/NumericColumnModule.cpp
@@ -80,7 +80,7 @@ void NumericColumnModuleImpl::addToContext(std::vector<std::vector<long>> & cont
if (index >= 0)
{
auto value = config.getAsFeature(column, index).get();
- try {res = value == "_" ? defaultValue : std::stof(value);}
+ try {res = (value == "_" or value == "NA") ? defaultValue : std::stof(value);}
catch (std::exception & e)
{util::myThrow(fmt::format("{} for '{}'", e.what(), value));}
}
--
GitLab