packages/tasks/src/tabular-classification/data.ts

import type { TaskDataCustom } from "../Types";

const taskData: TaskDataCustom = {
	datasets: [
		{
			description: "A comprehensive curation of datasets covering all benchmarks.",
			id: "inria-soda/tabular-benchmark",
		},
	],
	demo: {
		inputs: [
			{
				table: [
					["Glucose", "Blood Pressure ", "Skin Thickness", "Insulin", "BMI"],
					["148", "72", "35", "0", "33.6"],
					["150", "50", "30", "0", "35.1"],
					["141", "60", "29", "1", "39.2"],
				],
				type: "tabular",
			},
		],
		outputs: [
			{
				table: [["Diabetes"], ["1"], ["1"], ["0"]],
				type: "tabular",
			},
		],
	},
	metrics: [
		{
			description: "",
			id: "accuracy",
		},
		{
			description: "",
			id: "recall",
		},
		{
			description: "",
			id: "precision",
		},
		{
			description: "",
			id: "f1",
		},
	],
	models: [
		{
			description: "Breast cancer prediction model based on decision trees.",
			id: "scikit-learn/cancer-prediction-trees",
		},
	],
	spaces: [
		{
			description: "An application that can predict defective products on a production line.",
			id: "scikit-learn/tabular-playground",
		},
		{
			description: "An application that compares various tabular classification techniques on different datasets.",
			id: "scikit-learn/classification",
		},
	],
	summary: "Tabular classification is the task of classifying a target category (a group) based on set of attributes.",
	widgetModels: ["scikit-learn/tabular-playground"],
	youtubeId: "",
};

export default taskData;