Skip to content

Commit

Permalink
feat(models): add text classification models module and necessary cla…
Browse files Browse the repository at this point in the history
…sses
  • Loading branch information
entelecheia committed Mar 24, 2024
1 parent 977214b commit 484db0f
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions src/hyfiml/models/__init__.py
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",
]

0 comments on commit 484db0f

Please sign in to comment.