<?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);