【发布时间】:2020-11-06 17:03:18
【问题描述】:
我正在尝试为卫星图像分类的一些深度学习问题创建一个函数。我已经搜索了很多库,但我没有找到我的需求我尝试了这个 sikit-learn,但我觉得它不是我需要的
任何关于我可能看不到的特殊功能的提示?
【问题讨论】:
标签: python validation testing split training-data
我正在尝试为卫星图像分类的一些深度学习问题创建一个函数。我已经搜索了很多库,但我没有找到我的需求我尝试了这个 sikit-learn,但我觉得它不是我需要的
任何关于我可能看不到的特殊功能的提示?
【问题讨论】:
标签: python validation testing split training-data
sklearn train_test_split 似乎可以满足您的所有需求。
https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.train_test_split.html
【讨论】:
这应该可以解决问题。如果您愿意,可以分别对 X 和 y 数据使用置换数组。
num_tr, num_va = int(len(data)*0.5), int(len(data)*0.2)
perm = np.random.permutation(len(data))
tr_data = data[perm[:num_tr]]
va_data = data[perm[num_tr:num_tr+num_va]]
te_data = data[perm[num_tr+num_va:]]
【讨论】: