cii-text-analysis/save/train.php.save

27 lines
697 B
Text
Raw Permalink Normal View History

2026-02-10 16:19:59 +01:00
<?php
require_once 'vendor/autoload.php';
use Classifier\Lib\MessageQualifier;
require_once __DIR__ . '/Lib/MessageQualifier.php';
$keywordsAlertes = require_once __DIR__ . '/data/keywords.php';
$samples = [];
$labels = [];
$keywordsAlertes['ALERTE'] = array_merge($keywordsAlertes['ALERTE'], require_once __DIR__ . '/output/positive.php');
$keywordsAlertes['INFO'] = array_merge($keywordsAlertes['INFO'], require_once __DIR__ . '/output/negative.php');
foreach ($keywordsAlertes as $category => $messages) {
foreach ($messages as $msg) {
$samples[] = [$msg];
$labels[] = $category;
}
}
$classifier = new MessageQualifier();
$classifier->train($samples, $labels);