[swift] Swift Core ML을 사용하여 자동 번역을 수행하는 방법은 무엇인가요?

11 Dec 2023

swift

먼저, 번역 모델을 Core ML 형식으로 변환해야 합니다. 변환된 모델은 .mlmodel 파일로 저장됩니다. Core ML Tools를 사용하여 모델을 변환할 수 있습니다. 다음은 모델 변환의 예시입니다.

import CoreMLTools

let yourModel = YourTranslationModel()
try yourModel.write(toFile: "YourTranslationModel.mlmodel")

다음으로, iOS 프로젝트에 .mlmodel 파일을 추가합니다. Xcode에서 “Add Files to ‘YourProject’“를 선택하고 .mlmodel 파일을 추가합니다.

이제 Swift 코드에서 Core ML 모델을 사용하여 자동 번역을 수행할 수 있습니다. 아래는 예시 코드입니다.

import CoreML

// YourTranslationModel은 .mlmodel 파일로 생성된 모델 클래스입니다.
let model = YourTranslationModel()

func translate(text: String) -> String {
    if let prediction = try? model.prediction(text: text) {
        return prediction.translation
    } else {
        return "Translation failed"
    }
}

let inputText = "Hello, how are you?"
let translatedText = translate(text: inputText)
print(translatedText) // 출력: "안녕, 어떻게 지내세요?"

이와 같이 Core ML을 사용하여 Swift에서 자동 번역을 수행할 수 있습니다. 물론, 모델의 성능과 정확도는 모델의 품질에 따라 달라집니다.

참고 자료:

Core ML: https://developer.apple.com/documentation/coreml
Core ML Tools: https://developer.apple.com/documentation/coreml/core_ml_tools
Turi Create: https://github.com/apple/turicreate