Lstm in Python