packages/tasks/src/tabular-regression/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: [
					["Car Name", "Horsepower", "Weight"],
					["ford torino", "140", "3,449"],
					["amc hornet", "97", "2,774"],
					["toyota corolla", "65", "1,773"],
				],
				type: "tabular",
			},
		],
		outputs: [
			{
				table: [["MPG (miles per gallon)"], ["17"], ["18"], ["31"]],
				type: "tabular",
			},
		],
	},
	metrics: [
		{
			description: "",
			id: "mse",
		},
		{
			description:
				"Coefficient of determination (or R-squared) is a measure of how well the model fits the data. Higher R-squared is considered a better fit.",
			id: "r-squared",
		},
	],
	models: [
		{
			description: "Fish weight prediction based on length measurements and species.",
			id: "scikit-learn/Fish-Weight",
		},
	],
	spaces: [
		{
			description: "An application that can predict weight of a fish based on set of attributes.",
			id: "scikit-learn/fish-weight-prediction",
		},
	],
	summary: "Tabular regression is the task of predicting a numerical value given a set of attributes.",
	widgetModels: ["scikit-learn/Fish-Weight"],
	youtubeId: "",
};

export default taskData;