remove dependency on tensorflow
"""Executes model training and evaluation."""
-
import argparse
import logging
import sys
from sklearn import model_selection
-from tensorflow import gfile
from trainer import input_util
from trainer import model
-def _makedirs(dirname):
- """Similar to `mkdir -p` that works locally and on GCS."""
-
- if not gfile.Exists(dirname):
- gfile.MakeDirs(dirname)
-
-
def _train_and_evaluate(estimator, dataset, output_dir):
"""Runs model training and evalation."""
@@ -98,4 +89,3 @@ def main():
if __name__ == '__main__':
main()
-
