Mode | Name | Size | |
---|---|---|---|
-rw-r--r-- | BUILD | 5716 | logplain |
-rw-r--r-- | __init__.py | 1016 | logplain |
-rw-r--r-- | custom_export_strategy.py | 12645 | logplain |
-rw-r--r-- | custom_export_strategy_test.py | 9513 | logplain |
-rw-r--r-- | custom_loss_head.py | 3030 | logplain |
-rw-r--r-- | distillation_loss.py | 2952 | logplain |
-rw-r--r-- | dnn_tree_combined_estimator.py | 38672 | logplain |
-rw-r--r-- | dnn_tree_combined_estimator_test.py | 8578 | logplain |
-rw-r--r-- | estimator.py | 27650 | logplain |
-rw-r--r-- | estimator_test.py | 19787 | logplain |
-rw-r--r-- | estimator_utils.py | 2903 | logplain |
-rw-r--r-- | model.py | 15622 | logplain |
-rw-r--r-- | trainer_hooks.py | 9303 | logplain |
-rw-r--r-- | trainer_hooks_test.py | 3065 | logplain |