蓝桥云课xBilibili编程竞赛6月专场【1.金鸡独立】题解

qq_43335313

于 2022-07-05 22:02:03 发布

阅读量87

点赞数

分类专栏：机器学习入门文章标签： python 机器学习人工智能

本文链接：https://blog.csdn.net/qq_43335313/article/details/125597786

版权

机器学习入门专栏收录该内容

5 篇文章 0 订阅

订阅专栏

在这里插入图片描述

题解

from typing import List


def label_process(labels: List[str], sample_y: List[str])->List[List[int]]:
    # 在此函数中完成 one-hot 编码功能
    train_y = []
    LabelDict = dict(zip(labels,range(len(labels))))
    labelNum = len(labels)

    for i in sample_y:
        one_hot = [0]*labelNum
        one_hot[LabelDict[i]] = 1
        train_y.append(one_hot)
    return train_y
labels = ['Python', 'Java', 'Tensorflow', 'Springboot', 'Keras']
sample_y = ['Python', 'Python', 'Python', 'Java', 'Java', 'Keras']
print(label_process(labels,sample_y))