Antrenarea modelelor cu API-ul Estimators

Introducere în TensorFlow în Python

Isaiah Hull

Visiting Associate Professor of Finance, BI Norwegian Business School

Ce este API-ul Estimators?

Submodul de nivel înalt
Mai puțin flexibil
Impune bune practici
Deployment mai rapid
Multe modele predefinite

Această imagine prezintă o schemă a API-ului Estimators și relația sa cu API-urile de nivel inferior.

¹ Image taken from https://www.tensorflow.org/guide/premade_estimators

Specificarea și antrenarea modelului

Definirea coloanelor de caracteristici
Încărcarea și transformarea datelor
Definirea unui estimator
Aplicarea operației de antrenare

Definirea coloanelor de caracteristici

# Import tensorflow under its standard alias
import tensorflow as tf

# Define a numeric feature column
size = tf.feature_column.numeric_column("size")

# Define a categorical feature column
rooms = tf.feature_column.categorical_column_with_vocabulary_list("rooms",\
["1", "2", "3", "4", "5"])

Definirea coloanelor de caracteristici

# Create feature column list
features_list = [size, rooms]

# Define a matrix feature column
features_list = [tf.feature_column.numeric_column('image', shape=(784,))]

Încărcarea și transformarea datelor

# Define input data function
def input_fn():
    # Define feature dictionary
    features = {"size": [1340, 1690, 2720], "rooms": [1, 3, 4]}
    # Define labels
    labels = [221900, 538000, 180000]
    return features, labels

Definirea și antrenarea unui estimator de regresie

# Define a deep neural network regression
model0 = tf.estimator.DNNRegressor(feature_columns=feature_list,\
    hidden_units=[10, 6, 6, 3])

# Train the regression model
model0.train(input_fn, steps=20)

Definirea și antrenarea unei rețele neuronale profunde

# Define a deep neural network classifier
model1 = tf.estimator.DNNClassifier(feature_columns=feature_list,\ 
   hidden_units=[32, 16, 8], n_classes=4)

# Train the classifier
model1.train(input_fn, steps=20)

https://www.tensorflow.org/guide/estimators

Să exersăm!

Introducere în TensorFlow în Python