Added Softmax to Maths folder (#516)

This commit is contained in:
Fergus McDonald
2020-10-30 23:50:29 -07:00
committed by GitHub
parent 2190b92e1b
commit 9c9640c1c3
3 changed files with 27 additions and 0 deletions

View File

@@ -0,0 +1,12 @@
import { Softmax } from '../Softmax'
describe('Softmax', () => {
it('should return equal distribution of 1 for equal input values', () => {
expect(Softmax([1, 1])).toEqual([0.5, 0.5])
expect(Softmax([1, 1, 1, 1])).toEqual([0.25, 0.25, 0.25, 0.25])
})
it('should return values which sum to the value of 1', () => {
expect(Softmax([1, 2, 3, 4]).reduce((a, b) => a + b, 0)).toEqual(1)
})
})