Yogi set to retain government in UP