SpamMessage
Use the LSTM to train a spam message filter and convert the model to CoreML to predict on iPhone locally
Done:
- build model(use LinearVC model with the accuracy of 99%)
- convert the model to CoreML(.mlmodel)
- add the spam message extenion to the project
- Convert the orgin message to Tokenizer(implemented the ti-idf method in swift)
- according the predict result to filter the message to spam or ham
Spam message data are from SMS Spam Collection v. 1