import Cocoa
import CreateML
import Foundation
let filePath = "/Users/yamamotokyou/Desktop/epinions3.csv"
let outputFileName = "/Users/yamamotokyou/Desktop/SentimentAnalysisClassiferModel.mlmodel"
let dataURL = URL(fileURLWithPath: filePath)
let data = try MLDataTable(contentsOf: dataURL)
let (trainingData,testingData) = data.randomSplit(by: 0.8)
let sentimentClassifier = try MLTextClassifier(trainingData: trainingData, textColumn: "text", labelColumn: "class")
let metadata = MLModelMetadata(author: "Kyo Yamamoto", shortDescription: "Sentiment Analysis Model", version: "1.0")
try sentimentClassifier.write(to: URL(fileURLWithPath: outputFileName), metadata: metadata)
-
Notifications
You must be signed in to change notification settings - Fork 0
YamamotoDesu/SentimentAnalysisUsingPlaygrounds
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description or website provided.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published