Solved by Deathstroke_Twelved in post #8 
              
              
                Just fixed it - all I had to do was increase the number of epochs to an absurdly large number! 
That’s deep learning for you haha. Thanks everyone:)!