keras中sample_weight的使用

我会带着你远行 2021-11-01 04:48 543阅读 0赞

百度了好久，没有找到与sample\_weight相关的博客，于是自己摸索一下。  
sample\_weight是keras中的fit的参数，中文文档介绍如下：  
![在这里插入图片描述][20190811170826677.png]  
简单点的解释如下：参考https://blog.csdn.net/weixin\_40755306/article/details/82290033\#commentBox

![在这里插入图片描述][20190811171245849.png]  
sample\_weight的作用就是为数据集中的数据分配不同的权重。

我的例子是要将数据集的数据分为三类，用0，1，2代表这三类，我这里想为0分配权重0.3，为1分配权重1，为2分配权重2.。  
我的数据是存储在csv文件中的，我提取出标签列表，标签列表的内容是0，1，2的集合，列表名称为y\_train。我用下面代码生成一个权重的列表：

sample_weights=[]
    for sw in range(len(y_train)):
        if y_train[sw]==0:
                sample_weights.append(0.3)
        if y_train[sw]==1:
                sample_weights.append(1)
        if y_train[sw]==2:
                sample_weights.append(2)
    sample_weights=np.array(sample_weights)

当标签为0时， sample\_weights添加0.3，当标签为1时， sample\_weights添加1，当标签为2时， sample\_weights添加2。  
这里记得不要漏了最后一行，将列表转化为numpy数组。因为sample\_weight只能是numpy数组。  
创建好数组之后，下一步是要在compile中添加一个参数，先看看是添加哪个参数：  
![在这里插入图片描述][20190811174423241.png]  
这里的sample\_weight\_mode分为两种形式，如果你的权重形式是像我这样的，就是1D，那sample\_weight\_mode就设置为None。2D的形式还没试过，但如果用2D形式，那sample\_weight\_mode就要设置为sample\_weight\_mode=‘temporal’ 。

compile设置完就要设置fit了，我的模型有两个输出，但是我只想设置分类输出，我这里的分类输出层命名为’classifier’。那就在fit中添加一个参数：

sample_weight={'classifier' : sample_weights}

sample\_weights是我们上面定义的数组。

这样便可简单实现对数据的加权。

[20190811170826677.png]: /images/20211101/a273eee031034fb685cc826227e09ec8.png
[20190811171245849.png]: /images/20211101/31cdfd02c25e4a4c8fff8207ccd894ef.png
[20190811174423241.png]: /images/20211101/bec5742489494eba849ec5dc09462206.png