Merge pull request #682 from jimthompson5802/tf2_sequence_feature

TF2 porting: sequence feature
ludwig-ai · Apr 20, 2020 · ff51198 · ff51198
2 parents a3a09b1 + 4578a92
commit ff51198
Show file tree

Hide file tree

Showing 7 changed files with 799 additions and 999 deletions.
diff --git a/ludwig/features/numerical_feature.py b/ludwig/features/numerical_feature.py
@@ -37,7 +37,7 @@
 
 logger = logging.getLogger(__name__)
 
-
+# TODO TF2 can we eliminate use of these customer wrapper classes?
 # custom class to handle how Ludwig stores predictions
 class MSELoss(MeanSquaredError):
     def __init__(self, **kwargs):