Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2025-01-07 01:48:16 +01:00
Code Issues Packages Projects Releases Wiki Activity

Updated trainer_utils.py

Browse Source
This commit is contained in:
Utsav Dave 2023-04-04 08:40:49 -04:00 committed by GitHub
parent b5bba18856
commit 393938d7f4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
twml/twml/contrib/trainers/trainer_utils.py
View File

@ -56,7 +56,7 @@ def build_keras_trainer(
metrics_fn: metrics function used by TensorFlow estimators. metrics_fn: metrics function used by TensorFlow estimators.
Defaults to twml.metrics.get_binary_class_metric_fn(). Defaults to twml.metrics.get_binary_class_metric_fn().
**kwargs: for people familar with twml Trainer's options, they can be passed in here **kwargs: for people familiar with twml Trainer's options, they can be passed in here
as kwargs, and they will be forwarded to Trainer as opts. as kwargs, and they will be forwarded to Trainer as opts.
See https://cgit.twitter.biz/source/tree/twml/twml/argument_parser.py#n43 for available args. See https://cgit.twitter.biz/source/tree/twml/twml/argument_parser.py#n43 for available args.