-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(models): add text classification models module and necessary cla…
…sses
- Loading branch information
1 parent
977214b
commit 484db0f
Showing
1 changed file
with
26 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,26 @@ | ||
""" | ||
This module provides the necessary classes for text classification models. | ||
Classes: | ||
- TextClassifier: A class representing a text classifier model. | ||
- DatasetConfig: A class representing the configuration for the dataset used in training. | ||
- TrainingConfig: A class representing the configuration for the training process. | ||
- CrossValidateConfig: A class representing the configuration for cross-validation. | ||
Usage: | ||
Import the desired class from this module to use it in your code. | ||
""" | ||
|
||
from .text_classifier import ( | ||
CrossValidateConfig, | ||
DatasetConfig, | ||
TextClassifier, | ||
TrainingConfig, | ||
) | ||
|
||
__all__ = [ | ||
"TextClassifier", | ||
"DatasetConfig", | ||
"TrainingConfig", | ||
"CrossValidateConfig", | ||
] |