/
🤖

Basic TensorFlow Setup

PythonMachine LearningTensorFlow

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
Code Snippets & Notes by Makzan