AI를 사용한 스팸 이메일 분류기 구축: 기본 애플리케이션

npm install natural

const natural = require('natural');

// Create a new Naive Bayes classifier
const classifier = new natural.BayesClassifier();

// Sample spam and non-spam data
const spamData = [
  { text: "Congratulations, you've won a 00 gift card!", label: 'spam' },
  { text: "You are eligible for a free trial, click here to sign up.", label: 'spam' },
  { text: "Important meeting tomorrow at 10 AM", label: 'not_spam' },
  { text: "Let's grab lunch this weekend!", label: 'not_spam' }
];

// Add documents to the classifier (training data)
spamData.forEach(item => {
  classifier.addDocument(item.text, item.label);
});

// Train the classifier
classifier.train();

// Function to classify an email
function classifyEmail(emailContent) {
  const result = classifier.classify(emailContent);
  return result === 'spam' ? "This is a spam email" : "This is not a spam email";
}

// Example of using the classifier to detect spam
const testEmail = "Congratulations! You have won a 00 gift card.";
console.log(classifyEmail(testEmail)); // Output: "This is a spam email"

// Save the trained model to a file (optional)
classifier.save('spamClassifier.json', function(err, classifier) {
  if (err) {
    console.log('Error saving classifier:', err);
  } else {
    console.log('Classifier saved successfully!');
  }
});

node spamClassifier.js

This is a spam email
Classifier saved successfully!

const natural = require('natural');

// Load the saved classifier
natural.BayesClassifier.load('spamClassifier.json', null, function(err, classifier) {
  if (err) {
    console.log('Error loading classifier:', err);
  } else {
    // Classify a new email
    const testEmail = "You have won a free iPhone!";
    console.log(classifier.classify(testEmail)); // Output: 'spam' or 'not_spam'
  }
});