Improve conditions and refactor dataset classes (#475)

* Reimplement conditions

* Refactor datasets and implement LabelBatch

---------

Co-authored-by: Dario Coscia <dariocos99@gmail.com>
This commit is contained in:
Filippo Olivo
2025-03-07 11:24:09 +01:00
committed by Nicola Demo
parent bdad144461
commit a0cbf1c44a
40 changed files with 943 additions and 550 deletions

View File

@@ -1,12 +1,41 @@
"""
Module for conditions.
"""
__all__ = [
"Condition",
"ConditionInterface",
"DomainEquationCondition",
"InputPointsEquationCondition",
"InputOutputPointsCondition",
"InputTargetCondition",
"TensorInputTensorTargetCondition",
"TensorInputGraphTargetCondition",
"GraphInputTensorTargetCondition",
"GraphInputGraphTargetCondition",
"InputEquationCondition",
"InputTensorEquationCondition",
"InputGraphEquationCondition",
"DataCondition",
"GraphDataCondition",
"TensorDataCondition",
]
from .condition_interface import ConditionInterface
from .condition import Condition
from .domain_equation_condition import DomainEquationCondition
from .input_equation_condition import InputPointsEquationCondition
from .input_output_condition import InputOutputPointsCondition
from .input_target_condition import (
InputTargetCondition,
TensorInputTensorTargetCondition,
TensorInputGraphTargetCondition,
GraphInputTensorTargetCondition,
GraphInputGraphTargetCondition,
)
from .input_equation_condition import (
InputEquationCondition,
InputTensorEquationCondition,
InputGraphEquationCondition,
)
from .data_condition import (
DataCondition,
GraphDataCondition,
TensorDataCondition,
)