con - machine learning java example

Alojando un modelo weka a un código java

Question

He guardado el resultado de la clasificación weka haciendo clic derecho en el modelo y seleccionando "guardar modelo". Ahora, quiero cargarlo y trabajar en mi aplicación "JAVA". ¿Cómo puedo hacer eso? Los modelos podrían ser sesgo ingenuo, árbol de decisión y regresión. Necesito usar estos tres modelos.

Cualquier sugerencia o solución sería apreciada.

Gracias.

Answer 1

Si guardó un modelo en un archivo en WEKA, puede usarlo leyendo el objeto Java generado. Aquí hay un ejemplo con el clasificador Random Forest (previamente guardado en un archivo en WEKA):

RandomForest rf = (RandomForest) (new ObjectInputStream(PATH_TO_MODEL_FILE)).readObject();

No te olvides de las importaciones:

import weka.core.Instance;
import weka.core.Instances;
import weka.core.Attribute;
import weka.core.FastVector;
import weka.classifiers.trees.RandomForest;

Answer 2

Aquí hay un ejemplo, suponiendo que tiene un modelo RandomTree guardado en un archivo model.weka (cambie a cualquier clasificador y archivo que tenga)

RandomTree treeClassifier = (RandomTree) SerializationHelper.read(new FileInputStream("model.weka")));