Wie verwende ich SMOTE in der Java Weka API?

8

Ich versuche, ein Klassifizierungsmodell mit der Java Weka API zu erstellen. Mein Trainingsdatensatz weist Probleme mit dem Klassenungleichgewicht auf. Aus diesem Grund möchte ich SMOTE verwenden, um das Problem des Klassenungleichgewichts zu verringern. Ich weiß jedoch nicht, wie ich es in der Java Weka API verwenden soll.

Sagor Ali
quelle

Antworten:

8

Willkommen in der Community.

Sie können den folgenden Code verwenden:

import weka.filters.supervised.instance.SMOTE;
import weka.core.Instances;
import weka.core.converters.ConverterUtils.DataSource;
import weka.filters.Filter;

Instances data = DataSource.read(".../file.arff"); //Dataset 

SMOTE smote=new SMOTE();  //create object of SMOTE
smote.setInputFormat(data);
Instances data_smote = Filter.useFilter(data, smote); //Apply SMOTE on Dataset
Rejaul Karim
quelle
Danke für deine Unterstützung. Ich versuche, Ihren Code anzuwenden.
Sagor Ali
Ich freue mich zu hören, dass.
Rejaul Karim