Установите зависимости, включая библиотеку @tensorflow/tfjs-node, используя следующую команду в вашем терминале:

java

npm install @tensorflow/tfjs-node
Создайте файл model.ts и импортируйте необходимые зависимости:
typescript

import * as tf from '@tensorflow/tfjs-node';
Создайте функцию createModel, которая будет принимать одно число и возвращать 2 числа после softmax:
typescript

export function createModel(): tf.LayersModel {
  const model = tf.sequential();

  // Добавляем входной слой
  model.add(tf.layers.dense({
    inputShape: [1],
    units: 1,
    activation: 'relu'
  }));

  // Добавляем выходной слой
  model.add(tf.layers.dense({
    units: 2,
    activation: 'softmax'
  }));

  // Компилируем модель
  model.compile({
    loss: 'categoricalCrossentropy',
    optimizer: tf.train.adam(),
    metrics: ['accuracy']
  });

  return model;
}

Экспортируйте функцию createModel для дальнейшего использования:

typescript


export default createModel;
Готово! Вы можете использовать эту модель для предсказания 2 чисел после softmax.
ติดตั้งการขึ้นต่อกัน รวมถึงไลบรารี @tensorflow/tfjs-node โดยใช้คำสั่งต่อไปนี้ในเทอร์มินัลของคุณ: java npm ติดตั้ง @tensorflow/tfjs-node สร้างไฟล์ model.ts และนำเข้าการขึ้นต่อกันที่จำเป็น: typescript import * as tf from ' @tensorflow/tfjs- โหนด'; สร้างฟังก์ชัน createModel ที่จะรับหนึ่งหมายเลขและส่งกลับ 2 หมายเลขหลังจาก softmax: typescript ฟังก์ชันการส่งออก createModel(): tf.LayersModel { const model = tf.sequential(); // เพิ่มเลเยอร์อินพุต model.add(tf.layers.dense({ inputShape: [1], หน่วย: 1, การเปิดใช้งาน: 'relu' })); // เพิ่มเลเยอร์เอาต์พุต model.add(tf.layers.dense({ หน่วย: 2, การเปิดใช้งาน: 'softmax' })); // รวบรวมโมเดล model.compile({ loss: 'categoricalCrossentropy', เครื่องมือเพิ่มประสิทธิภาพ: tf.train.adam(), ตัวชี้วัด: ['accuracy'] }); แบบจำลองการคืน; } ส่งออกฟังก์ชัน createModel เพื่อใช้ในภายหลัง: ค่าเริ่มต้นการส่งออก typescript createModel; พร้อม! คุณสามารถใช้แบบจำลองนี้ทำนายตัวเลข 2 ตัวหลัง softmax
spam fails_task lang_mismatch pii not_appropriate hate_speech sexual_content quality toxicity humor helpfulness creativity violence