Я изучаю курс Udacity и столкнулся с некоторыми проблемами.В приведенном ниже коде, если я уберу опцию «axis = 0» в сумме в функции softmax, то же число будет напечатано, но показан совершенно другой график.Кто-нибудь имеет какие-либо идеи, почему это происходит?
Вот код:
scores = [3.0, 1.0, 0.2]
import numpy as np
def softmax(x):
return(np.exp(x)/np.sum(np.exp(x), axis=0))
print(softmax(scores))
# Plot softmax curves
import matplotlib.pyplot as plt
x = np.arange(-2.0, 6.0, 0.1)
scores = np.vstack([x, np.ones_like(x), 0.2 * np.ones_like(x)])
plt.plot(x, softmax(scores).T, linewidth=2)
plt.show();
Ценю любую помощь по этому вопросу.