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
Uh oh!
There was an error while loading. Please reload this page.
-
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