Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Biên dịch mô hình đã export với TensorFlow


TensorFlow (Google) framework học máy mã nguồn mở. Cài đặt: pip install tensorflow. Tensor: mảng đa chiều với Rank, Type, Shape.

Sử dụng Illiad dataset: 3 bản dịch (Cowper, Derby, Butler). Phân loại người dịch. Đã tiền xử lý.

Chạy trên Google Colab (Python trên trình duyệt, GPU miễn phí).

Ví dụ

print("Biên dịch exported model")
export_model.compile(
    loss=losses.SparseCategoricalCrossentropy(from_logits=False),
    optimizer='adam',
    metrics=['accuracy'])

Nguồn mã: TensorFlow Text Loading Tutorial

Đầu ra

Biên dịch exported model

Giải thích

  • Sau khi export mô hình, biên dịch bằng compile().
  • Loss: SparseCategoricalCrossentropy (from_logits=False vì đã có activation sigmoid).
  • Optimizer: Adam. Metric: accuracy.