AI — TensorFlow

What is TensorFlow?

Open-source machine learning framework by Google.

Install

pip install tensorflow

Basic Model

import tensorflow as tf

# Create model
model = tf.keras.Sequential([
    tf.keras.layers.Dense(128, activation='relu'),
    tf.keras.layers.Dense(10, activation='softmax')
])

# Compile
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

# Train
model.fit(X_train, y_train, epochs=10)

Layers

Layer	Description
Dense	Fully connected
Conv2D	Convolutional
LSTM	Recurrent
Dropout	Regularization

Mini Practice

1. Install TensorFlow
2. Build simple model
3. Train and evaluate
4. Try different layers

Up Next

Continue with PyTorch — PyTorch basics.