This version can save all weight matrix and predict labels to txt type.