Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Tensorflow를 Estimator와 함께 사용하여 훈련된 모델에서 예측을 수행하려면 어떻게 해야 합니까?

<시간/>

Tensorflow는 estimator와 함께 사용되어 'classifier' 방법에 있는 'predict' 방법을 사용하여 새 데이터에 대한 출력을 예측할 수 있습니다.

자세히 알아보기: TensorFlow란 무엇이며 Keras가 TensorFlow와 함께 신경망을 생성하는 방법은 무엇입니까?

모든 레이어에 정확히 하나의 입력 텐서와 하나의 출력 텐서가 있는 일반 레이어 스택으로 작업하는 데 사용되는 순차 모델을 구축하는 데 도움이 되는 Keras Sequential API를 사용할 것입니다.

하나 이상의 레이어를 포함하는 신경망을 컨볼루션 레이어라고 합니다. Convolutional Neural Network를 사용하여 학습 모델을 구축할 수 있습니다.

TensorFlow Text에는 TensorFlow 2.0과 함께 사용할 수 있는 텍스트 관련 클래스 및 작업 모음이 포함되어 있습니다. TensorFlow Text는 시퀀스 모델링을 전처리하는 데 사용할 수 있습니다.

Google Colaboratory를 사용하여 아래 코드를 실행하고 있습니다. Google Colab 또는 Colaboratory는 브라우저를 통해 Python 코드를 실행하는 데 도움이 되며 구성이 필요 없고 GPU(그래픽 처리 장치)에 대한 무료 액세스가 필요합니다. Colaboratory는 Jupyter Notebook을 기반으로 구축되었습니다.

Estimator는 완전한 모델에 대한 TensorFlow의 상위 수준 표현입니다. 손쉬운 확장 및 비동기식 교육을 위해 설계되었습니다.

모델은 홍채 데이터 세트를 사용하여 학습됩니다. 4개의 기능과 하나의 레이블이 있습니다.

  • 꽃받침 길이
  • 꽃받침 너비
  • 꽃잎 길이
  • 꽃잎 너비

print(“Generating predictions from model”)
expected = ['Setosa', 'Versicolor', 'Virginica']
predict_x = {
   'SepalLength': [5.1, 5.9, 6.9],
   'SepalWidth': [3.3, 3.0, 3.1],
   'PetalLength': [1.7, 4.2, 5.4],
   'PetalWidth': [0.5, 1.5, 2.1],
}
print(“Defining input function for prediction”)
print(“It converts inputs to dataset without labels”)
def input_fn(features, batch_size=256):
return tf.data.Dataset.from_tensor_slices(dict(features)).batch(batch_size)
predictions = classifier.predict(
   input_fn=lambda: input_fn(predict_x))

코드 크레딧 −https://www.tensorflow.org/tutorials/estimator/premade#first_things_first

출력

Generating predictions from model
Defining input function for prediction
It converts inputs to dataset without labels

설명

  • 학습된 모델은 좋은 결과를 생성합니다.
  • 이것은 레이블이 지정되지 않은 특정 측정값을 기반으로 붓꽃의 종을 예측하는 데 사용할 수 있습니다.
  • 예측은 단일 함수 호출을 사용하여 이루어집니다.