import prisma from "@/lib/prisma";

async function main() {
  console.log("Sedang memulai proses seeding...");

  const modelData = [
    {
      modelName: "Model XGBoost (Baseline)",
      description:
        "Model awal menggunakan parameter default XGBoost (learning_rate=0.3, max_depth=6) pada dataset yang tidak seimbang.",
      accuracy: 0.8,
      macroF1: 0.56,
      f1Negative: 0.61,
      f1Neutral: 0.16,
    },
    {
      modelName: "Model XGBoost (Tuned)",
      description:
        "Model dengan optimasi Hyperparameter menggunakan Grid Search untuk mencari kombinasi learning_rate dan max_depth terbaik.",
      accuracy: 0.81,
      macroF1: 0.58,
      f1Negative: 0.65,
      f1Neutral: 0.17,
    },
    {
      modelName: "Model XGBoost (Optimized)",
      description:
        "Model final menggunakan teknik SMOTE untuk menyeimbangkan kelas, seleksi fitur Chi-Square, dan optimasi Grid Search.",
      accuracy: 0.82,
      macroF1: 0.61,
      f1Negative: 0.65,
      f1Neutral: 0.27,
    },
  ];

  for (const data of modelData) {
    const model = await prisma.model.create({
      data: data,
    });
    console.log(`Berhasil membuat model: ${model.modelName}`);
  }

  console.log("Proses seeding selesai.");
}

main()
  .catch((e) => {
    console.error(e);
    process.exit(1);
  })
  .finally(async () => {
    await prisma.$disconnect();
  });