sparknlp_jsl._tf_graph_builders.graph_builders.AssertionTFGraphBuilder#

class sparknlp_jsl._tf_graph_builders.graph_builders.AssertionTFGraphBuilder(build_params)[source]#

Bases: TFGraphBuilder

Class to build the the TF graphs for AssertionDLApproach

Examples

>>> from sparknlp_jsl.training import tf_graph
>>> from sparknlp_jsl.base import *
>>> from sparknlp.annotator import *
>>> from sparknlp_jsl.annotator import *
>>> from sparknlp_jsl.annotator import *
>>>feat_size = 200
>>>n_classes = 6
>>> tf_graph.build("assertion_dl",build_params={"n_classes": n_classes}, model_location= "./tf_graphs", model_filename="blstm_34_32_30_{}_{}.pb".format(feat_size, n_classes))
>>> assertion = AssertionDLApproach() \
>>>               .setLabelCol("label") \
>>>               .setInputCols("document", "chunk", "embeddings") \
>>>               .setOutputCol("assertion") \
>>>               .setBatchSize(128) \
>>>               .setDropout(0.1) \
>>>               .setLearningRate(0.001) \
>>>               .setEpochs(50) \
>>>               .setValidationSplit(0.2) \
>>>               .setStartCol("start") \
>>>               .setEndCol("end") \
>>>               .setMaxSentLen(250) \
>>>               .setEnableOutputLogs(True) \
>>>               .setOutputLogsPath('training_logs/') \
>>>               .setGraphFolder('tf_graphs')

Methods

__init__(build_params)
build(model_location, model_filename)
check_build_params()
get_build_param(build_param)
get_build_params()
get_build_params_with_defaults()
get_model_build_param_explanations()
get_model_build_params()
get_model_filename()
supports_auto_file_name()