Skip to content

Commit

Permalink
added fnn_demo to layers so I can finish my example
Browse files Browse the repository at this point in the history
example is finished and works just like the benchmark model
  • Loading branch information
Gerhardsa0 committed Nov 15, 2023
1 parent ad75240 commit 28f5863
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
3 changes: 2 additions & 1 deletion src/safeds/ml/nn/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Layers for neural networks tasks."""

from ._rnn_layer import RNN_Layer
from ._model import Model
from ._model import Model
from ._fnn_layer import FNN_Layer
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
from torch.utils.data import DataLoader
from safeds.data.tabular.containers import Column, Table, TaggedTable, TimeSeries
from safeds.exceptions import ColumnSizeError, DuplicateColumnNameError
from safeds.ml.nn import RNN_Layer, Model
from safeds.ml.nn import FNN_Layer,RNN_Layer, Model



Expand All @@ -24,9 +24,9 @@ def test_create_timeseries() -> None:
hidden_dim = 256
output_dim = ts._forecast_horizon
layer1 = RNN_Layer(input_dim, hidden_dim)
layer2 = RNN_Layer(hidden_dim, output_dim)
layer2 = FNN_Layer(hidden_dim, output_dim)
model = Model([layer1, layer2])
#model.train(ts.into_DataLoader(), 5, 0.01)
model.train(ts.into_train_DataLoader(0.75), 50, 0.001)



Expand Down

0 comments on commit 28f5863

Please sign in to comment.