最近机器学习课有个作业是实现softmax多分类鸢尾花数据集,之前从来没推过softmax的公式,直接拿来用了,好好研究了一下,发现这个原理的推导还是有不少复杂的东西,分享一下结果,公式比较复杂,直接上手写了。
根据推导的结果,利用numpy手动实现了基于Iris数据集的softmax多分类,准确率有97.77%
import numpy as np
from sklearn import datasets
from matplotlib import pyplot as plt
class softmax_classify:
def __init__(self,lr=0.1,ep
1