Form Recognizer의 Custom model을 python을 이용해 가져오기

구 버전(2022-08-31 이)에서 model 가져오기

from azure.core.credentials import AzureKeyCredential
from azure.ai.formrecognizer import FormTrainingClient

# 구독 키와 엔드포인트 설정
endpoint = "FORM_RECOGNIZER_ENDPOINT"
key = "FORM_RECOGNIZER_KEY"

# AzureKeyCredential 생성
credential = AzureKeyCredential(subscription_key)

# FormTrainingClient 인스턴스 생성
training_client = FormTrainingClient(endpoint, credential)

# 계정의 모델에 대한 정보 가져오기
account_properties = training_client.get_account_properties()
print("custom model count = {}".format(account_properties.custom_model_count))

# 사용자 정의 모델 목록 가져오기
models = training_client.list_custom_models()

modellist = {}

# 각 모델의 정보 출력
for model in models:
    print("Model ID:", model.model_id)
    print("Status:", model.status)
    print("Created:", model.created_on)
    print("Last trained:", model.last_updated_on)
    modellist[model.model_id] = model.training_completed_on

print(modellist)

 

 

 

custom model 은 4개 있으나,

정보를 가져올 수 없다.

 

왜일까?

 

azure.ai.formrecognizer.FormTrainingClient class

 

FormTrainingClient 는 2022-08-31 이전버전에서만 사용해야 하는구나..

나는 2022-08-31 버전을 쓰고 있으니까 DocumentModelAdministrationClient 를 써야 하는구나…

 

 

 

 

 

 

azure.ai.formrecognizer.DocumentModelAdministrationClient class

 

신 버전(2022-08-31 이후)에서 model 가져오기

from azure.core.credentials import AzureKeyCredential  
from azure.ai.formrecognizer import DocumentModelAdministrationClient

# 구독 키와 엔드포인트 설정
endpoint = "FORM_RECOGNIZER_ENDPOINT"
key = "FORM_RECOGNIZER_KEY"

def getModel():
    # DocumentModelAdministrationClient 인스턴스 생성
    document_model_admin_client = DocumentModelAdministrationClient(endpoint=endpoint, credential=AzureKeyCredential(key))  
		
    # 사용자 정의 모델 목록 가져오기
    models = document_model_admin_client.list_document_models()  

    modellist = {}
    # 각 모델의 정보 출력
    for model in models:  
        modellist[model.model_id] = model.created_on
        print("{} | {} | {}".format(model.model_id, model.description, model.created_on))  

    return modellist

if __name__ == "__main__":
    getModel()

prebuilt 를 빼면 4건이 맞으니,

custom model 을 잘 가져왔다.