Commit b6ebb965 authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

TFTS: Add a head config option to StructuralEnsembleRegressor 

Adds a unit test for OneShotHead, fiddles with the train_op to deal with a dtype error the new unit tests uncovered.

PiperOrigin-RevId: 203179477
parent d04622b5
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment