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:
committed by
Nicola Demo
parent
bdad144461
commit
a0cbf1c44a
@@ -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,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user