With the gradual popularity of active magnetic bearings, it takes a lot of manpower and time to deal with the different control parameters in different fields and applications. Therefore, this research proposes to use machine learning techniques to realize the unbalance force rejection control parameter in the active magnetic bearing control automatically, which is completed by the program. Then try to discuss the feasibility of this method.