专栏首页算法与编程之美Python基础 | 快速实现label_to_index

Python基础 | 快速实现label_to_index

在做机器学习分类问题的时候,经常需要将标签名转换为对应的索引。什么是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', 'tulips']

接下来通过列表推导式遍历数组,然后再将列表转换为字典。其中enumerate()可以获得下标。

label_to_index = dict([(name, index) for index, name in enumerate(class_names)])

有没有更简洁的方式呢?答案就是利用下面的列表推导式更简洁的实现。

label_to_index = {name:index for index,name in enumerate(class_names)}

本文为大家介绍了利用列表推导式快速的实习label_to_index,这一机器学习中最常用的操作。

本文分享自微信公众号 - 算法与编程之美(algo_coding),作者:gongsuochen

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2019-07-24

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

我来说两句

0 条评论
登录 后参与评论

相关文章

  • Html|Vue实战小项目-购物车

    在很多电商网站中,都有一个非常重要的功能,那就是购物车。接下来将使用Vue.js实现一个简易的购物车项目。实现的功能有全选或选择部分商品、计算总价、对商品数量进...

    算法与编程之美
  • 微信小程序|API音频与视频组件的插入使用

    1、音频组件控制 首先在微信小程序中插入音频组件需首先引入一个audioContext对象,之后再通过 audioId 跟一个audio组件绑定,通过它可以操作...

    算法与编程之美
  • 前端|BootStrap 布局组件

    <span class=“glyphicon glyphicon-home”></span>

    算法与编程之美
  • 初窥 Elasticsearch-PHP [1.0]

    Elasticsearch-PHP 下载地址:http://www.thinkphp.cn/code/download/id/1290.html 教程以Th...

    双面人
  • 排查 Node.js 服务内存泄漏,没想到竟是它?

    团队最近将两个项目迁移至 degg 2.0 中,两个项目均出现比较严重的内存泄漏问题,此处以本人维护的埋点服务为例进行排查。服务上线后内存增长如下图,其中红框为...

    五月君
  • 循环中的异步&&循环中的闭包

    foo函数下的index输出5,全局下的index不存在 现在我们把var 换为let

    陌上寒
  • Java 使用Spring-data-elasticsearch 为索引添加别名 修改及替换别名

    一直感觉spring-data-es封装的很迷..调用起来奇奇怪怪 还是封装一层方便自己调用吧

    heasy3
  • C++核心准则C.132:不要没有理由就将函数声明为虚函数‍

    C.132: Don't make a function virtual without reason

    面向对象思考
  • Todo List: 新增编辑任务,localStorage + Vuex实现本地存储 – 第二章

    在第一章中,我们的Todo List已经初现锥形,本章节我们对页面优化,同时让数据实现本地存储。主要用到localStorage + Vuex来实现页面交互和存...

    Javanx
  • MAVEN简介之——pom.xml

    maven是围绕着构建生命周期这个核心概念为基础的。maven里有3个内嵌的构建生命周期,default,clean和site。

    小忽悠

扫码关注云+社区

领取腾讯云代金券