🤖

Basic TensorFlow Setup

July 21, 2021

The example code was originally from DeepLearning.ai. It shows the very basic setup to get started using TensorFlow.

import tensorflow as tf
import numpy as np

xs = np.array([1.0,2,3,4,5,6])
ys = np.array([1,1.5,2,2.5,3,3.5])

model = tf.keras.Sequential([
  tf.keras.layers.Dense(units=1, input_shape=[1])
])

model.compile(optimizer='sgd', loss='mean_squared_error')
model.fit(xs, ys, epochs=500)

print(model.predict([7.0]))

Edit this page