0

0

如何利用 JavaScript 实现一个简单的机器学习模型进行预测或分类?

狼影

狼影

发布时间:2025-10-06 18:15:02

|

580人浏览过

|

来源于php中文网

原创

答案是JavaScript可实现简单机器学习模型。通过手动实现线性回归和kNN算法,可在前端完成基础预测与分类任务;结合TensorFlow.js则能训练神经网络,支持更复杂场景,适合轻量级应用开发。

如何利用 javascript 实现一个简单的机器学习模型进行预测或分类?

用 JavaScript 实现一个简单的机器学习模型是完全可行的,尤其适合初学者理解基本原理或在前端进行轻量级预测。虽然不像 Python 那样有丰富的库支持,但借助一些工具和算法,可以在浏览器中完成分类或预测任务。

使用简单线性回归进行数值预测

线性回归是最基础的机器学习模型之一,适用于预测连续值(如房价、温度等)。我们可以用最小二乘法手动实现一个一元线性回归模型。

假设我们有一组数据点 (x, y),目标是找到一条直线 y = mx + b 来最好地拟合这些点。

以下是实现步骤:

立即学习Java免费学习笔记(深入)”;

  • 计算 x 和 y 的平均值
  • 利用公式求出斜率 m 和截距 b
  • 用训练好的模型对新输入进行预测

代码示例:

function linearRegression(x, y) {
  const n = x.length;
  let sumX = 0, sumY = 0, sumXY = 0, sumXX = 0;
<p>for (let i = 0; i < n; i++) {
sumX += x[i];
sumY += y[i];
sumXY += x[i] * y[i];
sumXX += x[i] ** 2;
}</p><p>const m = (n <em> sumXY - sumX </em> sumY) / (n * sumXX - sumX *<em> 2);
const b = (sumY - m </em> sumX) / n;</p><p>return { slope: m, intercept: b };
}</p><p>// 示例数据:房屋面积(平方米)与价格(万元)
const areas = [50, 60, 70, 80, 90];
const prices = [100, 120, 140, 160, 180];</p><p>const model = linearRegression(areas, prices);</p><p>// 预测 75 平方米的房子价格
const prediction = model.slope * 75 + model.intercept;
console.log(<code>预测价格: ${prediction} 万元</code>);</p>

使用 k-近邻(kNN)进行分类

kNN 是一种简单的分类算法,基于“物以类聚”的思想。给定一个新样本,找出训练集中最接近它的 k 个点,然后根据它们的类别投票决定预测结果。

实现要点:

腾讯交互翻译
腾讯交互翻译

腾讯AI Lab发布的一款AI辅助翻译产品

下载
  • 定义距离函数(如欧氏距离)
  • 对每个测试样本,计算它与所有训练样本的距离
  • 选出最近的 k 个邻居
  • 统计这 k 个邻居中最多的类别作为预测结果

代码示例:

function euclideanDistance(point1, point2) {
  return Math.sqrt(point1.reduce((sum, val, i) => sum + (val - point2[i]) ** 2, 0));
}
<p>function knnClassify(trainingData, labels, testPoint, k = 3) {
const distances = trainingData.map((point, idx) => ({
dist: euclideanDistance(testPoint, point),
label: labels[idx]
}));</p><p>distances.sort((a, b) => a.dist - b.dist);</p><p>const kLabels = distances.slice(0, k).map(d => d.label);
const labelCounts = {};</p><p>kLabels.forEach(label => {
labelCounts[label] = (labelCounts[label] || 0) + 1;
});</p><p>let predictedLabel = null;
let maxCount = 0;
for (const label in labelCounts) {
if (labelCounts[label] > maxCount) {
maxCount = labelCounts[label];
predictedLabel = label;
}
}</p><p>return predictedLabel;
}</p><p>// 示例:根据身高体重判断性别(1=男,0=女)
const data = [
[180, 75],
[170, 65],
[160, 50],
[165, 55],
[175, 70]
];
const labels = ['男', '男', '女', '女', '男'];</p><p>const result = knnClassify(data, labels, [168, 60], 3);
console.log(<code>预测性别: ${result}</code>);</p>

使用 TensorFlow.js 进行更复杂的模型训练

如果想在浏览器中运行更强大的模型,可以使用 TensorFlow.js。它支持构建神经网络、训练模型,并直接在网页中运行。

例如,使用 TensorFlow.js 训练一个简单的线性回归模型:

import * as tf from '@tensorflow/tfjs';
<p>async function trainModel() {
// 准备数据
const xs = tf.tensor2d([50, 60, 70, 80, 90], [5, 1]);
const ys = tf.tensor2d([100, 120, 140, 160, 180], [5, 1]);</p><p>// 创建模型
const model = tf.sequential();
model.add(tf.layers.dense({ units: 1, inputShape: [1] }));</p><p>// 编译模型
model.compile({ optimizer: 'sgd', loss: 'meanSquaredError' });</p><p>// 训练模型
await model.fit(xs, ys, { epochs: 100 });</p><p>// 预测
const prediction = model.predict(tf.tensor2d([75], [1, 1]));
prediction.print(); // 输出预测值
}</p><p>trainModel();</p>

这种方式更适合处理复杂任务,比如图像识别、文本分类等,而且能利用 GPU 加速。

基本上就这些。从手动实现简单算法到使用 TensorFlow.js 构建深度学习模型,JavaScript 完全可以胜任轻量级机器学习任务。关键是理解模型原理,再选择合适的工具实现。不复杂但容易忽略细节,比如数据归一化、过拟合等问题,在实际项目中需要注意。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
js正则表达式
js正则表达式

php中文网为大家提供各种js正则表达式语法大全以及各种js正则表达式使用的方法,还有更多js正则表达式的相关文章、相关下载、相关课程,供大家免费下载体验。

530

2023.06.20

js获取当前时间
js获取当前时间

JS全称JavaScript,是一种具有函数优先的轻量级,解释型或即时编译型的编程语言;它是一种属于网络的高级脚本语言,主要用于Web,常用来为网页添加各式各样的动态功能。js怎么获取当前时间呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

576

2023.07.28

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

760

2023.08.03

js是什么意思
js是什么意思

JS是JavaScript的缩写,它是一种广泛应用于网页开发的脚本语言。JavaScript是一种解释性的、基于对象和事件驱动的编程语言,通常用于为网页增加交互性和动态性。它可以在网页上实现复杂的功能和效果,如表单验证、页面元素操作、动画效果、数据交互等。

6207

2023.08.17

js删除节点的方法
js删除节点的方法

js删除节点的方法有:1、removeChild()方法,用于从父节点中移除指定的子节点,它需要两个参数,第一个参数是要删除的子节点,第二个参数是父节点;2、parentNode.removeChild()方法,可以直接通过父节点调用来删除子节点;3、remove()方法,可以直接删除节点,而无需指定父节点;4、innerHTML属性,用于删除节点的内容。

492

2023.09.01

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

221

2023.09.04

Js中concat和push的区别
Js中concat和push的区别

Js中concat和push的区别:1、concat用于将两个或多个数组合并成一个新数组,并返回这个新数组,而push用于向数组的末尾添加一个或多个元素,并返回修改后的数组的新长度;2、concat不会修改原始数组,是创建新的数组,而push会修改原数组,将新元素添加到原数组的末尾等等。本专题为大家提供concat和push相关的文章、下载、课程内容,供大家免费下载体验。

240

2023.09.14

js截取字符串的方法介绍
js截取字符串的方法介绍

JavaScript字符串截取方法,包括substring、slice、substr、charAt和split方法。这些方法可以根据具体需求,灵活地截取字符串的不同部分。在实际开发中,根据具体情况选择合适的方法进行字符串截取,能够提高代码的效率和可读性 。

303

2023.09.21

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

76

2026.03.11

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
React 教程
React 教程

共58课时 | 6万人学习

TypeScript 教程
TypeScript 教程

共19课时 | 3.4万人学习

Bootstrap 5教程
Bootstrap 5教程

共46课时 | 3.6万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号