1. Job Roadmap
  2. Home
  3. Problem Set
  4. codenotelist
  5. Forum
  6. course
  7. Shore Share Sessions
  8. Record
  1. Login
  2. Sign Up
  3. Language
    1. English
    2. 한국어
    3. 简体中文
    4. 正體中文
    ZhContent TextSol AI分析

解题思路

按照题意一步一步实现即可。

  1. 数据读取与转换

    • 从标准输入读取 JSON 对象,解析出训练集 train 和测试集 test。
    • 转换为 numpy 数组方便后续矩阵运算。

P3344.第2题-均方误差

    1000ms Tried: 100 Accepted: 29 Difficulty: 3 所属公司 : 美团
    算法与标签>数学

题目内容

给定一批训练样本与若干测试样本,请你手写实现主成分分析 (PCA)(PCA)(PCA) 并仅保留第一主成分来压缩-重建数据,最后输出每个测试样本在重建后的均方误差 (MSE)(MSE)(MSE) 。

1.输入读取

  • traintraintrain - 二维列表,每行是一个 mmm 维数值特征向量

  • testtesttest - 二维列表,维度同上

2.去均值 (mean−center)(mean-center)(mean−center)

Xc=X−μ,μ=mean(Xtrain)X_c=X-μ,μ=mean(X_{train})Xc​=X−μ,μ=mean(Xtrain​)

3.协方差矩阵(总体方差,ddof=0ddof=0ddof=0)

∑=1nXcTXc∑=\frac {1}{n}X^T_cX_c∑=n1​XcT​Xc​ 。

4.求第一主成分

  • 用 numpy.linalg.eighnumpy.linalg.eighnumpy.linalg.eigh 得到全部特征对 (λi,νi)(λ_i,ν_i)(λi​,νi​)

  • 按特征值从大到小选取第一主成分 νmaxν_{max}νmax​

  • 方向标准化规则 -- 若 νmaxν_{max}νmax​ 首个非零分量为负,则整体乘以 −1-1−1 ;这样方向唯一

5.投影-重建

z=(x−μ)Tνmax,z=(x-μ)^Tνmax,z=(x−μ)Tνmax, x^=μ+z\hat{x}=μ+zx^=μ+z νmaxνmaxνmax

6.输出

  • 对每个测试样本计算

MSE(x)=1m∑j=1m(xj−x^j)2MSE(x)=\frac {1}{m}\sum^m_{j=1}(x_j-\hat{x}_j)^2MSE(x)=m1​∑j=1m​(xj​−x^j​)2

  • 结果保留两位小数,使用字符串形式

  • 所有测试样本的误差按输入顺序组成 JSON 数组一行输出

输入描述

标准输入仅一行,为如下 JSONJSONJSON 对象:

{

"train":[[...],[...],...],

"test":[[...],[...],...]

}

其中

  • traintraintrain 长度 n≥2n≥2n≥2 ,每行长度 m≥2m≥2m≥2

  • testtesttest 任意条数,维度同 traintraintrain

输出描述

标准输出仅一行 -- 测试集中 每个样本 MSEMSEMSE 的字符串形式(两位小数),用 JSONJSONJSON 数组包裹.

补充说明

1.算法不含随机过程(无需设置随机种子)。

2.为了确保通过所有测试用例,仅允许使用 NumpyNumpyNumpy库与PandasPandasPandas库实现本题。

3.使用总体方差 np.var(x,ddof=0)np.var(x,ddof=0)np.var(x,ddof=0).

样例1

输入

{"train": [[0,0],[0,1],[1,0],[1,1]],"test":[[0.5,0.5],[1.5,1.5]]}

输出

["0.00","0.50"]

登录后即可使用 AI 分析。

模式
倒计时时长
:

最长 10 小时 59 分;应用后按此时长重新开始。

提示:点击提交记录在左侧题面区域查看详情
题库
AI分析设置
留空使用官方API Key,每天有次数限制(自定义API Key仅限会员和管理员使用,不限次数)
会员和管理员可切换模型;切到 Kimi/智谱/通义/豆包时需填写对应供应商 API Key
升级会员,可将运行与提交冷却时间缩短至 1 秒起

Status

  • Judging Queue
  • Service Status

Development

  • Open Source

Support

  • Help
  • Contact Us

About

  • About
  • Privacy
  • Terms of Service
  • Copyright Complaint
  1. Language
    1. English
    2. 한국어
    3. 简体中文
    4. 正體中文
  2. Legacy mode
  3. Theme
    1. Light
    2. Dark
  1. 京ICP备2025123107号-1
  2. Worker 1, 59ms
  3. Powered by Hydro v5.0.0-beta.18 Community
CLOSE


ScanQRCodePrompt

请使用微信扫描下方二维码完成注册

Forgot password or username?