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 { ...@@ -32,6 +32,7 @@ public class ConfigurationSymbol extends CommonScopeSpanningSymbol {
private LossSymbol loss; private LossSymbol loss;
private RewardFunctionSymbol rlRewardFunctionSymbol; private RewardFunctionSymbol rlRewardFunctionSymbol;
private NNArchitectureSymbol trainedArchitecture; private NNArchitectureSymbol trainedArchitecture;
private NNArchitectureSymbol criticNetwork;
public static final ConfigurationSymbolKind KIND = new ConfigurationSymbolKind(); public static final ConfigurationSymbolKind KIND = new ConfigurationSymbolKind();
...@@ -81,6 +82,14 @@ public class ConfigurationSymbol extends CommonScopeSpanningSymbol { ...@@ -81,6 +82,14 @@ public class ConfigurationSymbol extends CommonScopeSpanningSymbol {
this.trainedArchitecture = trainedArchitecture; this.trainedArchitecture = trainedArchitecture;
} }
public Optional<NNArchitectureSymbol> getCriticNetwork() {
return Optional.ofNullable(criticNetwork);
}
public void setCriticNetwork(NNArchitectureSymbol criticNetwork) {
this.criticNetwork = criticNetwork;
}
public Map<String, EntrySymbol> getEntryMap() { public Map<String, EntrySymbol> getEntryMap() {
return entryMap; 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