编写transform函数的问题

来源:4-8 scikit-learn中的Scaler

慕仔1219081

2022-08-02

transform函数中的assert,您讲的是Xtrain列数要与mean列数相同,可以写Xtrain列数与方差列数相同吗,是同理的吗

写回答

1回答

liuyubobobo

2022-08-02

可以。本质其实就是要求,传入 transform 中的 X,应该和 X 的列数相同。


在 fit 中,mean 和 scale 都是靠传来的 X 算出来的,且长度都和 X 的列数相同,所以在 transform 中,assert 保证 X 和 mean 的列数相同就够了(就一定和 scale 的列数相同)。你写成 assert 保证 X 和 scale 的列数相同也可以(就一定和 mean 的列数相同。)


继续加油!:)

0
0

Python3入门机器学习 经典算法与应用  

Python3+sklearn,兼顾原理、算法底层实现和框架使用。

5893 学习 · 2454 问题

查看课程