Browse Source

[FIX] minor changes

pull/6/head
troyyyyy 1 year ago
parent
commit
609fa36f1e
2 changed files with 3 additions and 3 deletions
  1. +1
    -1
      ablkit/learning/basic_nn.py
  2. +2
    -2
      ablkit/learning/model_converter.py

+ 1
- 1
ablkit/learning/basic_nn.py View File

@@ -54,7 +54,7 @@ class BasicNN:
in the ``fit`` and ``train_epoch`` methods. Defaults to None. in the ``fit`` and ``train_epoch`` methods. Defaults to None.
test_transform : Callable[..., Any], optional test_transform : Callable[..., Any], optional
A function/transform that takes an object and returns a transformed version in the A function/transform that takes an object and returns a transformed version in the
``predict``, ``predict_proba`` and ``score`` methods, . Defaults to None.
``predict``, ``predict_proba`` and ``score`` methods.Defaults to None.
collate_fn : Callable[[List[T]], Any], optional collate_fn : Callable[[List[T]], Any], optional
The function used to collate data. Defaults to None. The function used to collate data. Defaults to None.
""" """


+ 2
- 2
ablkit/learning/model_converter.py View File

@@ -71,7 +71,7 @@ class ModelConverter:
in the `fit` and `train_epoch` methods. Defaults to None. in the `fit` and `train_epoch` methods. Defaults to None.
test_transform : Callable[..., Any], optional test_transform : Callable[..., Any], optional
A function/transform that takes an object and returns a transformed version in the A function/transform that takes an object and returns a transformed version in the
`predict`, `predict_proba` and `score` methods, . Defaults to None.
`predict`, `predict_proba` and `score` methods.Defaults to None.
collate_fn : Callable[[List[T]], Any], optional collate_fn : Callable[[List[T]], Any], optional
The function used to collate data. Defaults to None. The function used to collate data. Defaults to None.


@@ -161,7 +161,7 @@ class ModelConverter:
in the `fit` and `train_epoch` methods. Defaults to None. in the `fit` and `train_epoch` methods. Defaults to None.
test_transform : Callable[..., Any], optional test_transform : Callable[..., Any], optional
A function/transform that takes an object and returns a transformed version in the A function/transform that takes an object and returns a transformed version in the
`predict`, `predict_proba` and `score` methods, . Defaults to None.
`predict`, `predict_proba` and `score` methods.Defaults to None.
collate_fn : Callable[[List[T]], Any], optional collate_fn : Callable[[List[T]], Any], optional
The function used to collate data. Defaults to None. The function used to collate data. Defaults to None.




Loading…
Cancel
Save