How to add custom dataset for the TabNet? #3038
adinath1233
started this conversation in
Development
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
TabularDataset dataset = loadCsv("/pathtodataset/dataset.csv");
private static TabularDataset loadCsv(String filePath) throws IOException {
// Load CSV file
CsvDataset dataset = CsvDataset.builder()
.optCsvFile(Paths.get(filePath))
.setSampling(100, true)
.setCsvFormat(CSVFormat.DEFAULT.withHeader("Geography", "Gender", "Age", "Tenure", "Balance", "NumberOfProducts", "HasCrCard", "ActiveMembers", "EstimatedSalary", "Exited", "CreditScore"))
.build();
I am trying to load the dataset like this but I am getting error "Missing Features"
Correct me if I am wrong is this a correct way to input?
Beta Was this translation helpful? Give feedback.
All reactions