Abstract: A method for predicting fault occurrence in a mechanical system and an electrical system. The method comprises: receiving a first dataset of mechanical system condition data, the first dataset being imbalanced by having more data points in a first category than in a second category; generating a plurality of chromosomes from the second category data points in the first dataset; the plurality of chromosomes including information to enable the creation of new datasets; generating a second dataset using the plurality of chromosomes and an evolutionary algorithm, the second dataset being less imbalanced than the first dataset; and predicting fault occurrence in the mechanical system using the second dataset and a machine learning algorithm.