diff --git a/torch_modules/src/NumericColumnModule.cpp b/torch_modules/src/NumericColumnModule.cpp
index 29138d5ece2849406c7f41f6e24713cdcd7b4076..4899d5953c908cfe51373864d95ad6c788e0827c 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));}
       }