sklearn数据集的获取与划分

获取sklearn本地的数据集

from sklearn.datasets import load_iris
li = load_iris()

print("数据集描述为:")
print(li.DESCR)
print("目标值为:")
print(li.target)
print("数据为:")
print(li.data)
print("特征描述名称为:")
print(li.feature_names)
print("目标描述名为:")
print(li.target_names)

从网络获取数据集

from sklearn.datasets import fetch_20newsgroups

# 从网络获取大的数据集
news = fetch_20newsgroups(subset="all")
print("打印所有获取的数据:")
print(news.data) 

划分训练集和测试集

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
# 加载数据集
li = load_iris()

# 将数据划分为训练集特征值,训练集目标值, 测试集特征值, 测试集目标值
train_data,test_data,train_target,test_target = train_test_split(li.data, li.target, test_size = 0.25)

print("训练集特征值数据:")
print(train_data)
print("训练集目标值数据:")
print(train_target)
print("测试集特征值数据:")
print(test_data)
print("测试值目标值数据:")
print(test_target)

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏AI研习社

手把手教你在浏览器中使用脸部识别软件包 face-api.js

我可以很激动地说,我们终于有可能在浏览器中运行人脸识别程序了!在这篇文章中,我会给大家介绍一个基于 TensorFlow.js 核心的 JavaScript 模...

20310
来自专栏AI科技大本营的专栏

TensorSpace:一套用于构建神经网络3D可视化应用的框架

今天要为大家推荐一套超酷炫的,用于构建神经网络 3D 可视化应用的框架——TensorSpace。

12610
来自专栏机器之心

资源 | 微软开源MMdnn:实现多个框架之间的模型转换

选自GitHub 作者:Kit CHEN等 机器之心编译 参与:路雪、思源 近日,微软开源 MMdnn,可用于转换、可视化和诊断深度神经网络模型的全面、跨框架解...

38060
来自专栏Python中文社区

基于RNN自动生成古诗

專 欄 ❈ 作者:yonggege,Python中文社区专栏作者 GitHub地址:https://github.com/wzyonggege ❈ 0. ch...

29250
来自专栏人工智能LeadAI

TensorFlow从0到1丨 第五篇:TensorFlow轻松搞定线性回归

上一篇 第一个机器学习问题 其实是一个线性回归问题(line regression),呈现了用数据来训练模型的具体方式。本篇从平行世界返回,利用TensorFl...

36470
来自专栏机器学习算法与理论

调用Dlib库进行人脸关键点标记

       昨天调试了人脸识别(classifier_webcam)这个程序,效果不错,响应速度也挺快。按照http://blog.csdn.net/u011...

48190
来自专栏PPV课数据科学社区

R语言中不能进行深度学习?

摘要: R语言现在能也进行深度学习了,而且和python一样好,快来试一试吧。 众所周知,R语言是统计分析最好用的语言。但在Keras和TensorFlow的帮...

45890
来自专栏北京马哥教育

深度学习 + OpenCV,Python实现实时目标检测

使用 OpenCV 和 Python 上对实时视频流进行深度学习目标检测是非常简单的,我们只需要组合一些合适的代码,接入实时视频,随后加入原有的目标检测功能。 ...

1.1K70
来自专栏机器之心

教程 | 深度学习 + OpenCV,Python实现实时视频目标检测

选自PyimageSearch 机器之心编译 参与:路雪、李泽南 使用 OpenCV 和 Python 对实时视频流进行深度学习目标检测是非常简单的,我们只需要...

65170
来自专栏新智元

谷歌发布 TensorFlow Fold,支持动态计算图,GPU 增速 100 倍

【新智元导读】谷歌官方博客最新发布TensorFlow Fold,通过为每个输入构建单独的计算图解决由于输入的大小和结构不同导致的问题。此外,通过动态批处理,实...

39590

扫码关注云+社区

领取腾讯云代金券