o
    Si                     @   s\   d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl	m
  mZ dddZdS )zN
Define the function to create lexical analysis model and model's data reader
    N)NormalInitializertrainc                 C   sZ   t jdddgddd}t jdddgddd}|dkr+tj|| |dd	d
}|g||dS tS )zcreate lac modelwords   int64)nameshapedtype	lod_leveltargetsinferTN)	for_infertarget)	feed_listr   
crf_decode)fluiddatanetslex_netret)
vocab_size
num_labelsmoder   r   r    r   K/home/ubuntu/.local/lib/python3.10/site-packages/jieba/lac_small/creator.pycreate_model   s   
r   )r   )__doc__sysosmathpaddlepaddle.fluidr   paddle.fluid.initializerr   jieba.lac_small.nets	lac_smallr   r   r   r   r   r   <module>   s   