Commit 18344e12 authored by Nicola Gatto's avatar Nicola Gatto
Browse files

Add critic network to configuration symbol

parent e69c5e50
......@@ -32,6 +32,7 @@ public class ConfigurationSymbol extends CommonScopeSpanningSymbol {
private LossSymbol loss;
private RewardFunctionSymbol rlRewardFunctionSymbol;
private NNArchitectureSymbol trainedArchitecture;
private NNArchitectureSymbol criticNetwork;
public static final ConfigurationSymbolKind KIND = new ConfigurationSymbolKind();
......@@ -81,6 +82,14 @@ public class ConfigurationSymbol extends CommonScopeSpanningSymbol {
this.trainedArchitecture = trainedArchitecture;
}
public Optional<NNArchitectureSymbol> getCriticNetwork() {
return Optional.ofNullable(criticNetwork);
}
public void setCriticNetwork(NNArchitectureSymbol criticNetwork) {
this.criticNetwork = criticNetwork;
}
public Map<String, EntrySymbol> getEntryMap() {
return entryMap;
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment