Maison >interface Web >js tutoriel >Création d'un classificateur de courrier indésirable à l'aide de l'IA : une application de base
Ce projet utilise Node.js et la bibliothèque Natural pour créer une application basée sur l'IA qui classe les e-mails comme spam ou non spam. L'application utilise un classificateur Naive Bayes pour la détection du spam, qui est un algorithme courant pour les tâches de classification de texte.
Avant de commencer, assurez-vous que les éléments suivants sont installés :
mkdir spam-email-classifier cd spam-email-classifier
npm init -y
Exécutez la commande suivante pour installer les dépendances requises :
npm install natural
Créez un nouveau fichier JavaScript (par exemple, spamClassifier.js) et ajoutez le code suivant :
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!'); } });
Pour exécuter le classificateur, ouvrez un terminal et accédez au dossier du projet. Ensuite, exécutez la commande suivante :
node spamClassifier.js
Vous devriez voir un résultat similaire à celui-ci :
This is a spam email Classifier saved successfully!
Vous pouvez charger le modèle de classificateur plus tard pour classer les nouveaux e-mails. Voici comment charger le modèle et classer les nouveaux e-mails :
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' } });
Pour améliorer la précision du classificateur de spam, vous pouvez :
Si vous souhaitez envoyer ou recevoir des e-mails depuis l'application, vous pouvez utiliser la bibliothèque Nodemailer pour envoyer des e-mails.
mkdir spam-email-classifier cd spam-email-classifier
npm init -y
Ce guide vous a expliqué la configuration d'une application d'IA utilisant Node.js et Naive Bayes pour classer les e-mails comme spam ou non. Vous pouvez développer cette application en :
Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!