Part of Speech for Thai

Description

A Part of Speech classifier predicts a grammatical label for every token in the input text. Implemented with an averaged perceptron architecture.

Predicted Entities

  • NN
  • VV
  • PS
  • NG
  • NU
  • CL
  • PU
  • CC
  • AX
  • AV
  • FX
  • AJ
  • PR
  • PA
  • IJ
  • XX

Live Demo Open in Colab Download Copy S3 URI

How to use


document_assembler = DocumentAssembler() \
  .setInputCol("text") \
  .setOutputCol("document")

sentence_detector = SentenceDetector() \
  .setInputCols(["document"]) \
  .setOutputCol("sentence")

pos = PerceptronModel.pretrained("pos_lst20", "th") \
  .setInputCols(["document", "token"]) \
  .setOutputCol("pos")

pipeline = Pipeline(stages=[
  document_assembler,
  sentence_detector,
  posTagger
])

example = spark.createDataFrame([['สวัสดีจาก John Snow Labs! ']], ["text"])

result = pipeline.fit(example).transform(example)



val document_assembler = DocumentAssembler()
        .setInputCol("text")
        .setOutputCol("document")

val sentence_detector = SentenceDetector()
        .setInputCols("document")
.setOutputCol("sentence")

val pos = PerceptronModel.pretrained("pos_lst20", "th")
        .setInputCols(Array("document", "token"))
        .setOutputCol("pos")

val pipeline = new Pipeline().setStages(Array(document_assembler, sentence_detector, pos))

val data = Seq("สวัสดีจาก John Snow Labs! ").toDF("text")
val result = pipeline.fit(data).transform(data)


import nlu
text = [""สวัสดีจาก John Snow Labs! ""]
token_df = nlu.load('th.pos').predict(text)
token_df
    

Results

       token pos
                
0  สวัสดีจาก  CC
1       John  NN
2       Snow  NN
3       Labs  NN
4          !  PU

Model Information

Model Name: pos_lst20
Compatibility: Spark NLP 3.0.0+
License: Open Source
Edition: Official
Input Labels: [document, token]
Output Labels: [pos]
Language: th