吴恩达机器学习 课后作业 13.2 k-means pytorch实现
听说最近字节面试经常让手撕k-means代码,那不如就用pytorch撕一撕吧。之前学过的都忘得差不多了,参考了两个网上的代码,话不多说上代码import torchimport torch.nn as nnfrom scipy.io import loadmatimport numpy as nproot_path = '../machine-learning-ex7/ex7/'data_file_path = root_path + 'ex7data2.mat'data_fi.
原创
2021-03-24 21:01:45 ·
713 阅读 ·
1 评论