Train csv shape must be greater that test shape (automatize code)