Bagaimana cara menggunakan SMOTE di Java Weka API?

8

Saya mencoba membangun model klasifikasi menggunakan Java Weka API. Dataset pelatihan saya memiliki masalah ketidakseimbangan kelas. Untuk alasan ini, saya ingin menggunakan SMOTE untuk mengurangi masalah ketidakseimbangan kelas. Tapi, saya tidak tahu cara menggunakannya di Java Weka API.

Sagor Ali
sumber

Jawaban:

8

Selamat datang di komunitas.

Anda dapat menggunakan kode berikut:

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
sumber
Terima kasih atas dukunganmu. Saya mencoba menerapkan kode Anda.
Sagor Ali
Saya senang mendengarnya.
Rejaul Karim