LightAutoML release v0.3.8

lightautoml/ml_algo/dl_model.py

@@ -527,6 +527,7 @@ def get_dataloaders_from_dicts(self, data_dict: Dict):
                 num_workers=self.train_params["num_workers"],
                 collate_fn=collate_dict,
                 pin_memory=self.train_params["pin_memory"],
+                drop_last={'train':True}.get(stage, False)
             )
             for stage, value in data_dict.items()
         }

lightautoml/reader/base.py

@@ -505,7 +505,7 @@ def _guess_role(self, feature: Series) -> RoleType:
         try:
             # TODO: check all notnans and set coerce errors
             t = cast(pd.Series, pd.to_datetime(feature, infer_datetime_format=False, format=date_format))
-        except (ValueError, AttributeError):
+        except (ValueError, AttributeError, TypeError):
             # else category
             return CategoryRole(object)
 

lightautoml/text/trainer.py

@@ -233,7 +233,7 @@ def load_state_dict(self, weights: Dict, model: nn.Module):
             else:
                 model.load_state_dict(model_weights)
             self.models[key] = deepcopy(model.eval()).cpu()
-
+        weights["best_loss"] = self.best_loss
         return self
 
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "LightAutoML"
-version = "0.3.8-b1"
+version = "0.3.8"
 description = "Fast and customizable framework for automatic ML model creation (AutoML)"
 authors = [
     "Alexander Ryzhkov <[email protected]>",
@@ -15,11 +15,10 @@ license = "Apache-2.0"
 homepage = "https://lightautoml.readthedocs.io/en/latest/"
 repository = "https://github.com/AILab-MLTools/LightAutoML"
 classifiers = [
-    "Programming Language :: Python :: 3.6",
-    "Programming Language :: Python :: 3.7",
     "Programming Language :: Python :: 3.8",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
     "Operating System :: OS Independent",
     "Intended Audience :: Science/Research",
     "Development Status :: 3 - Alpha",
@@ -31,7 +30,7 @@ classifiers = [
 
 
 [tool.poetry.dependencies]
-python = ">=3.6.1, <3.11"
+python = ">=3.8, <3.12"
 
 poetry-core = [
     {version = ">=1.0.0", python = "<3.7"},

scripts/poetry_fix.py

@@ -10,8 +10,8 @@
 
 
 PYPROJECT_TOML = Path("pyproject.toml")
-ALL_PYTHON_DEPS = ">=3.6.1, <3.11"
-PYTHON_DEPS = {6: "~3.6.1", 7: "~3.7.1", 8: "~3.8.0", 9: "~3.9.0", 10: "~3.10.0"}
+ALL_PYTHON_DEPS = ">=3.6.1, <=3.11"
+PYTHON_DEPS = {6: "~3.6.1", 7: "~3.7.1", 8: "~3.8.0", 9: "~3.9.0", 10: "~3.10.0", 11: "~3.11.0"}
 PYTHON_DEPS_PATTERN = '^python = ".*"$'