Passa al contenuto principale

Tensorflow Estimator API

Prova Databricks gratis

Che cos'è l'API Tensorflow Estimator?

Gli stimatori rappresentano un modello completo ma, al tempo stesso, appaiono più intuitivi agli utenti meno esperti. L'API Estimator fornisce metodi per addestrare il modello, valutarne l'accuratezza e generare previsioni. TensorFlow offre uno stack di programmazione costituito da diversi livelli di API, come mostrato nell'immagine sottostante.

tensorflow estimators image

Esistono due tipi di stimatori: si possono scegliere stimatori predefiniti o, in alternativa, scrivere propri stimatori personalizzati. I modelli basati su stimatori possono essere eseguiti su host locali così come in un ambiente multiserver distribuito, senza cambiare il modello. Inoltre, i modelli basati su stimatori possono essere eseguiti su CPU, GPU o TPU senza dover registrare il modello.

Gli stimatori svolgono quattro funzioni principali:

  • Addestramento: addestrano un modello in base a un determinato input per un numero fisso di passaggi.
  • Valutazione: valutano il modello in base a una serie di test.
  • Previsione:  eseguono inferenze utilizzando il modello addestrato.
  • Esportazione del modello per serving.

In aggiunta, lo stimatore ha un comportamento standard comune ai lavori di addestramento, ad esempio salva e ripristina punti di controllo (checkpoint), crea sommari ecc. Uno stimatore richiede che l'utente scriva una model_fn e una input_fn che corrispondano alle porzioni di modello e input del grafo di TensorFlow.

Gli stimatori offrono numerosi vantaggi:

  • Semplificano le implementazioni di condivisione fra sviluppatori di modelli.
  • Consentono di sviluppare un modello eccellente con codice intuitivo di alto livello, poiché solitamente sono più facili da usare quando si devono creare modelli rispetto alle API di basso livello di TensorFlow.
  • Gli stimatori stessi sono basati su tf.keras.layers, agevolando notevolmente la personalizzazione.
  • Gli stimatori facilitano il lavoro dell'utente costruendo il grafo per suo conto.
  • Gli stimatori mettono a disposizione un ciclo di addestramento distribuito in modo sicuro che consente di controllare come e quando:
    • costruire il grafo
    • inizializzare le variabili
    • caricare i dati
    • gestire le eccezioni
    • creare file di punti di controllo e ripristinare il sistema in seguito a errori
    • salvare sommari per TensorBoard

 

Risorse aggiuntive

Torna al Glossario