deeplotx 0.4.12b6__tar.gz → 0.4.13__tar.gz

{deeplotx-0.4.12b6 → deeplotx-0.4.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deeplotx
-Version: 0.4.12b6
+Version: 0.4.13
 Summary: Easy-2-use long text NLP toolkit.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{deeplotx-0.4.12b6 → deeplotx-0.4.13}/deeplotx/nn/recursive_sequential.py

@@ -36,11 +36,10 @@ class RecursiveSequential(BaseNeuralNetwork):
         return x, (hidden_state, cell_state)
 
     @override
-    def predict(self, x, batch_size: int | None = None) -> torch.Tensor:
-        _batch_size = batch_size if batch_size is not None else x.shape[0]
+    def predict(self, x) -> torch.Tensor:
         __train = self.training
         self.training = False
         with torch.no_grad():
-            res = self.forward(x, self.initial_state(batch_size=_batch_size))[0]
+            res = self.forward(x.unsqueeze(0), self.initial_state(batch_size=1))[0]
         self.training = __train
         return res

{deeplotx-0.4.12b6 → deeplotx-0.4.13}/deeplotx.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deeplotx
-Version: 0.4.12b6
+Version: 0.4.13
 Summary: Easy-2-use long text NLP toolkit.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{deeplotx-0.4.12b6 → deeplotx-0.4.13}/deeplotx.egg-info/SOURCES.txt

@@ -27,5 +27,4 @@ deeplotx/trainer/base_trainer.py
 deeplotx/trainer/text_binary_classification_trainer.py
 deeplotx/util/__init__.py
 deeplotx/util/hash.py
-deeplotx/util/read_file.py
-test/test.py
+deeplotx/util/read_file.py

{deeplotx-0.4.12b6 → deeplotx-0.4.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "deeplotx"
-version = "0.4.12.beta6"
+version = "0.4.13"
 description = "Easy-2-use long text NLP toolkit."
 readme = "README.md"
 requires-python = ">=3.10"

deeplotx-0.4.12b6/test/test.py

@@ -1,21 +0,0 @@
-from deeplotx import TextBinaryClassifierTrainer, LongTextEncoder
-from deeplotx.util import get_files, read_file
-
-long_text_encoder = LongTextEncoder(
-    max_length=2048,
-    chunk_size=512,
-    overlapping=128
-)
-
-trainer = TextBinaryClassifierTrainer(
-    long_text_encoder=long_text_encoder,
-    batch_size=4,
-    train_ratio=0.9
-)
-
-pos_data_path = './data/pos'
-neg_data_path = './data/neg'
-pos_data = [read_file(x) for x in get_files(pos_data_path)]
-neg_data = [read_file(x) for x in get_files(neg_data_path)]
-model = trainer.train(pos_data, neg_data, num_epochs=20, learning_rate=2e-5, train_loss_threshold=1)
-model.predict(long_text_encoder.encode('这是一个测试文本.').squeeze())