ReviewClassifier.swift

ReviewClassifier.swift

import CoreML
import NaturalLanguage

class ReviewClassifier {
    let model: NLModel
    
    init() {
        let modelURL = Bundle.main.url(forResource: "ReviewModel", withExtension: "mlmodelc")!
        model = try! NLModel(contentsOf: modelURL)
    }
    
    func classifyReview(text: String) -> String {
        let classification = model.predictedLabel(for: text)
        print("Classification: \(classification)")
        return classification
    }
}