Learning device, estimating device, learning method, and computer program product
摘要:
A learning device includes one or more processors. The processors calculate a likelihood of belonging to a plurality of estimated classes, of learning data, by using an estimation model for estimating to which of the estimated classes input data belongs. The processors calculate a weight of a loss function to be used in learning the estimation model such that, when a likelihood of a first class that is closer to correct data than other estimated classes among the estimated classes and likelihoods of a second class and a third class that are adjacent to the first class are applied to a function having a predetermined shape, a position that has an extreme value of the function corresponds to the correct data. The processors learn the estimation model by using the loss function.
信息查询
0/0