본문 바로가기
WIKI 기술 지식 베이스

Train API

원문 보기 위키 갱신

출처: 문서

본문

Train API

train API 작업은 선택한 알고리즘을 기반으로 모델을 학습시켜요. 학습은 동기식과 비동기식으로 모두 수행할 수 있어요.

요청 예제

다음 예제는 k-means 알고리즘을 사용해 인덱스 데이터를 학습시켜요.

k-means로 동기식 학습하기

POST /_plugins/_ml/_train/kmeans
{
    "parameters": {
        "centroids": 3,
        "iterations": 10,
        "distance_type": "COSINE"
    },
    "input_query": {
        "_source": ["petal_length_in_cm", "petal_width_in_cm"],
        "size": 10000
    },
    "input_index": [
        "iris_data"
    ]
}

{% include copy-curl.html %}

k-means로 비동기식 학습하기

POST /_plugins/_ml/_train/kmeans?async=true
{
    "parameters": {
        "centroids": 3,
        "iterations": 10,
        "distance_type": "COSINE"
    },
    "input_query": {
        "_source": ["petal_length_in_cm", "petal_width_in_cm"],
        "size": 10000
    },
    "input_index": [
        "iris_data"
    ]
}

{% include copy-curl.html %}

응답 예제

동기식

동기식 응답의 경우 API는 model_id를 반환해요. 이 ID를 사용해 모델을 조회하거나 삭제할 수 있어요.

{
  "model_id" : "lblVmX8BO5w8y8RaYYvN",
  "status" : "COMPLETED"
}

비동기식

비동기식 응답의 경우 API는 task_id를 반환해요. 이 ID를 사용해 작업을 조회하거나 삭제할 수 있어요.

{
  "task_id" : "lrlamX8BO5w8y8Ra2otd",
  "status" : "CREATED"
}

더 알아보기 (Learn more)