IRIS: Towards an Accurate and Fast Stage Weight Prediction Method