One-hot编码Python实现（二）

最新推荐文章于 2024-06-26 11:35:37 发布

Bill_zhang5

最新推荐文章于 2024-06-26 11:35:37 发布

阅读量664

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/Bill_zhang5/article/details/88713002

版权

Python 专栏收录该内容

22 篇文章 1 订阅

订阅专栏

def dense_to_one_hot(labels_dense, num_classes):
    """Convert class labels from scalars to one-hot vectors."""
    
    num_labels = labels_dense.shape[0]
    index_offset = np.arange(num_labels) * num_classes
    labels_one_hot = np.zeros((num_labels, num_classes))
    labels_one_hot.flat[index_offset + labels_dense.ravel()] = 1
    return labels_one_hot