API Modules
generate_data()
visualize_data()
validate_dataset()
split_dataset()
train_pipeline()
evaluate_model()
TaskParser
TaskParser.parse()
DocumentBuilder
DocumentBuilder.build()
NERParser
NERParser.parse()
NERDocumentBuilder
NERDocumentBuilder.build()
TextCatParser
TextCatParser.parse()
TextCatDocumentBuilder
TextCatDocumentBuilder.build()
TaskHandler
TaskHandler.get_handler()
TaskHandler.supported_tasks()
DataGenerator
DataGenerator.__init__()
DataGenerator.run()
DataValidator
DataValidator.dataset_path
DataValidator.output_folder
DataValidator.task
DataValidator.nlp
DataValidator.json_path
DataValidator.png_path
DataValidator.__init__()
DataValidator.run()
DataVisualizer
DataVisualizer.input_path
DataVisualizer.task
DataVisualizer.n_samples
DataVisualizer.port
DataVisualizer.__init__()
DataVisualizer.run()
LLMClient
LLMClient.model
LLMClient.api_key
LLMClient.api_base
LLMClient.max_tokens
LLMClient.__init__()
LLMClient.generate()
LLMConfig
LLMConfig.model
LLMConfig.api_key
LLMConfig.api_base
LLMConfig.max_tokens
LLMConfig.model_config
load_llm_config()
PromptMessage
PromptMessage.role
PromptMessage.content
PromptMessage.model_config
PromptConfig
PromptConfig.system
PromptConfig.user
PromptConfig.model_config
NERExample
NERExample.text
NERExample.explanation
NERExample.model_config
NERStructuredConfig
NERStructuredConfig.task
NERStructuredConfig.entities
NERStructuredConfig.domain
NERStructuredConfig.tone
NERStructuredConfig.length
NERStructuredConfig.language
NERStructuredConfig.temperature
NERStructuredConfig.constraints
NERStructuredConfig.examples
NERStructuredConfig.validate_entities()
NERStructuredConfig.model_config
TextCatCategory
TextCatCategory.name
TextCatCategory.description
TextCatCategory.model_config
TextCatExample
TextCatExample.text
TextCatExample.category
TextCatExample.model_config
TextCatStructuredConfig
TextCatStructuredConfig.task
TextCatStructuredConfig.categories
TextCatStructuredConfig.domain
TextCatStructuredConfig.tone
TextCatStructuredConfig.length
TextCatStructuredConfig.language
TextCatStructuredConfig.temperature
TextCatStructuredConfig.constraints
TextCatStructuredConfig.examples
TextCatStructuredConfig.validate_categories()
TextCatStructuredConfig.model_config
load_prompt_config()
DataSpliter
DataSpliter.input_file
DataSpliter.train_file
DataSpliter.dev_file
DataSpliter.dev_size
DataSpliter.seed
DataSpliter.__init__()
DataSpliter.run()
ModelTrainer
ModelTrainer.train_data
ModelTrainer.base_model
ModelTrainer.output_model
ModelTrainer.n_iter
ModelTrainer.dropout
ModelTrainer.__init__()
ModelTrainer.run()
ModelEvaluater
ModelEvaluater.model_path
ModelEvaluater.eval_data
ModelEvaluater.__init__()
ModelEvaluater.run()
Please activate JavaScript to enable the search functionality.