AI ninja project [day 16] 文字处理 -- 回归

我们已经有了语音转文字的技术，
那我们也能将文字进行向量化。

那我们是否能收集客服人员顾客的回答，
然後将回应进行量化，
当作对客服人员考绩的参考呢?

这篇参考autokeras的文字回归攻略:
https://autokeras.com/tutorial/text_regression/

colab:
https://colab.research.google.com/github/keras-team/autokeras/blob/master/docs/ipynb/text_regression.ipynb

安装套件及载入:

一样使用IMDB影评的资料集，
不同於之前载入资料集的方法，这里采用sklearn的load_files来载入资料集
可以发现已经自动帮忙把pos及neg两个类别，转换成0或1的数值:

这里我们采用autokeras的TextRegressor来进行训练，并将测试集预测结果印出:

可以发现预测结果为介於0到1之前的数值。

官网也特别说明可以使用tensorflow dataset的资料格式用来训练: