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

@@ -46,8 +46,8 @@ class InverseDiffusionReactionProblem(
equation=Equation(diffusion_reaction),
),
"data": Condition(
input_points=LabelTensor(torch.randn(10, 2), ["x", "t"]),
output_points=LabelTensor(torch.randn(10, 1), ["u"]),
input=LabelTensor(torch.randn(10, 2), ["x", "t"]),
target=LabelTensor(torch.randn(10, 1), ["u"]),
),
}