GPT学习-使用embedding做回归(regression)预测

Regression"回归"是机器学习中的一个术语,指的是预测一个连续的数值,而不是分类的类别。这个词通常被翻译为"回归"。这个翻译有时候感觉有点晦涩难懂,也可以考虑使用"数值预测"或者"回归预测"理解上会更容易一些。

"回归"意味着预测一个数字,而不是其中的一个类别。我们将基于评论文本的嵌入来预测评分。我们将数据集分为训练集和测试集,以便于所有后续任务在未见过的数据上进行真实的性能评估。数据集是在Obtain_dataset.ipynb创建的。

这次学习的代码是,Regression_using_embeddings.ipynb

它目的是使用随机森林回归器(RandomForestRegressor)预测亚马逊评论的评分。

1.导入的库:sklearn.ensemble 中的 RandomForestRegressor 是用于回归预测的模型,sklearn.model_selection 中的 train_test_split 用于将数据集划分为训练集和测试集,sklearn.metrics 中的 mean_squared_error 和 mean_absolute_error 用于评估模型的预测性能。

2.数据划分:使用 train_test_split 函数将数据集划分为训练集和测试集。这里,80

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值