AR model stock forecast PYTHON