您当前的位置: 首页 >  Python

暂无认证

  • 0浏览

    0关注

    92582博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

Python基础 | 快速实现label_to_index

发布时间:2019-07-24 00:00:00 ,浏览量:0

欢迎点击「算法与编程之美」↑关注我们!

本文首发于微信公众号:"算法与编程之美",欢迎关注,及时了解更多此系列文章。

在做机器学习分类问题的时候,经常需要将标签名转换为对应的索引。什么是label_to_index问题?下面将通过一个案例来详细说明。

假设现在需要完成一个图像分类的任务,现有五种类型的花卉,分别是:['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips'],每个类别的花卉有图像若干。在做具体的图形分类任务之前,需要将这些标签名转换为对应的整型索引,这样的转换便于机器学习模型的后续处理。

我们的目标是建立一个字典类型的数据结构label_to_index,能够实现给定一个标签名,找到指定的数字索引。给大家一点时间,如果是你,如何快速的编码实现?

首先按照字典序对所有的标签名进行排序。

class_names = ['roses', 'sunflowers', 'daisy', 'dandelion', 'tulips']	
class_names = sorted(class_names)	
# ['daisy', 'dandelion', 'roses', 'sunflowers', 
关注
打赏
1653961664
查看更多评论
立即登录/注册

微信扫码登录

0.3551s