resources.wrappers.FileJsonKeras.gate-lf-keras-json.tests.example2.py Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of learningframework Show documentation

A GATE plugin that provides many different machine learning algorithms for a wide range of NLP-related machine learning tasks like text classification, tagging, or chunking.

There is a newer version: 4.2

Show newest version

import sys
sys.path.append("/Users/xingyi/Gate/learningFrameWorkDeepLearning/gate-lf-python-data/")
sys.path.append('/Users/xingyi/Gate/learningFrameWorkDeepLearning/gate-lf-keras-json/')
from gatelfdata import Dataset
from gatelfkerasjson import KerasWrapperImpl1
metaFile = sys.argv[1]
modelSavePrefix = sys.argv[2]
ds = Dataset(metaFile)

kerasModel = KerasWrapperImpl1(ds)
kerasModel.genKerasModel()
kerasModel.trainModel()
kerasModel.saveModel(modelSavePrefix)