/ fast_seqfunc / __init__.py
__init__.py
 1  """Top-level API for fast-seqfunc.
 2  
 3  This is the file from which you can do:
 4  
 5      from fast_seqfunc import train_model, predict, save_model, load_model
 6  
 7  Provides a simple interface for sequence-function modeling of proteins and nucleotides.
 8  """
 9  
10  from fast_seqfunc.core import (
11      evaluate_model,
12      load_model,
13      predict,
14      save_model,
15      train_model,
16  )
17  
18  # Import synthetic data generation functions
19  from fast_seqfunc.synthetic import (
20      create_classification_task,
21      create_g_count_task,
22      create_gc_content_task,
23      create_interaction_task,
24      create_length_dependent_task,
25      create_motif_count_task,
26      create_motif_position_task,
27      create_multiclass_task,
28      create_nonlinear_composition_task,
29      generate_dataset_by_task,
30  )
31  
32  __all__ = [
33      # Core functionality
34      "train_model",
35      "predict",
36      "save_model",
37      "load_model",
38      "evaluate_model",
39      # Synthetic data
40      "create_g_count_task",
41      "create_gc_content_task",
42      "create_motif_position_task",
43      "create_motif_count_task",
44      "create_length_dependent_task",
45      "create_nonlinear_composition_task",
46      "create_interaction_task",
47      "create_classification_task",
48      "create_multiclass_task",
49      "generate_dataset_by_task",
50  ]