Skip to content

GitLab

Commit 426b2c6e authored by Silas-Asamoah's avatar Silas-Asamoah Committed by Hongkun Yu
Browse files

Updated code to Tensorflow 2.0 (#7770)

parent aa031d91
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
samples/cookbook/regression/linear_regression.py
View file @ 426b2c6e
......@@ -22,6 +22,7 @@ import argparse
import numpy as np
import tensorflow as tf
from absl import app
import automobile_data
......@@ -98,5 +99,5 @@ def main(argv):
if __name__ == "__main__":
# The Estimator periodically generates "INFO" logs; make these logs visible.
tf.logging.set_verbosity(tf.logging.INFO)
tf.app.run(main=main)
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
app.run(main=main)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment