Class ContextClassifier
- java.lang.Object
-
- eu.h2020.helios_social.core.info_control.ContextClassifier
-
public class ContextClassifier extends java.lang.ObjectContextClassifier detects context of received message. It uses the @link MultiClassNaiveBayes classifier to classify message text content based on the training data from @link MessageContextDatabase.
-
-
Constructor Summary
Constructors Constructor Description ContextClassifier(MyContexts myContexts)ContextClassifier(MyContexts myContexts, int maxHistorySize)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description java.util.List<ContextProbability>classify(java.lang.String from, java.lang.String topic, java.lang.String text)Classifies a received message into context classes with probabilityvoidtrain(MessageContext messageContext)Trains the model with a MessageContext as a training samplevoidtrain(MessageContextRepository repository)Trains the model with all the MessageContexts in the training database (MessageContextDatabase)voidtrain(java.lang.String contextId, java.lang.String from, java.lang.String topic, java.lang.String text)Trains the model with contextual message information: from, topic and text
-
-
-
Constructor Detail
-
ContextClassifier
public ContextClassifier(MyContexts myContexts, int maxHistorySize)
-
ContextClassifier
public ContextClassifier(MyContexts myContexts)
-
-
Method Detail
-
train
public void train(MessageContextRepository repository)
Trains the model with all the MessageContexts in the training database (MessageContextDatabase)- Parameters:
repository- the MessageContextRepository
-
train
public void train(java.lang.String contextId, java.lang.String from, java.lang.String topic, java.lang.String text)Trains the model with contextual message information: from, topic and text- Parameters:
contextId-from-topic-text-
-
train
public void train(MessageContext messageContext)
Trains the model with a MessageContext as a training sample- Parameters:
messageContext-
-
classify
public java.util.List<ContextProbability> classify(java.lang.String from, java.lang.String topic, java.lang.String text)
Classifies a received message into context classes with probability- Parameters:
from-topic-text-- Returns:
-
-