Исправить случайное состояние для решения спиральной задачи с керасом

Question

Я пытаюсь воспроизвести результаты, показанные здесь https://glowingpython.blogspot.com/2017/04/solving-two-spirals-problem-with-keras.html, но я получаю другой результат / график в основном потому, что мне нужно исправить свое случайное состояние.Я пытался np.random.seed(), но это все еще не помогает мне.Я проверил, и говорят, что np.random.seed() должно быть достаточно, даже если я использую sklearn и matplotlib