import type { BuiltinDataset } from "@elvis/ui";

// Column indices start at 0
export const DATASETS: BuiltinDataset[] = [
  {
    name: "Iris",
    url: `${import.meta.env.BASE_URL}datasets/iris.csv`,
    inputColumns: [0, 1],
    targetColumn: 4,
  },
  {
    name: "Titanic",
    url: `${import.meta.env.BASE_URL}datasets/titanic.csv`,
    inputColumns: [0, 1],
    targetColumn: 5,
  },
];