Commit bb9fb3f8 authored by aa's avatar aa Committed by Julian Johannes Steinsberger-Dührßen

Added Tensorflow generator

parent 03696c7f
......@@ -85,6 +85,12 @@
<groupId>de.monticore.lang.monticar</groupId>
<artifactId>cnnarch-gluon-generator</artifactId>
<version>${cnnarch-gluon-generator.version}</version>
</dependency>
<dependency>
<groupId>de.monticore.lang.monticar</groupId>
<artifactId>cnnarch-tensorflow-generator</artifactId>
<version>${cnnarch-tensorflow-generator.version}</version>
</dependency>
<dependency>
......
......@@ -43,6 +43,16 @@ public enum Backend {
@Override
public CNNTrainGenerator getCNNTrainGenerator() {
return new CNNTrain2Gluon(new RewardFunctionCppGenerator());
}
},
TENSORFLOW{
@Override
public CNNArchGenerator getCNNArchGenerator() {
return new CNNArch2Tensorflow();
}
@Override
public CNNTrainGenerator getCNNTrainGenerator() {
return new CNNTrain2Tensorflow();
}
},
TENSORFLOW{
......@@ -72,6 +82,9 @@ public enum Backend {
case "TENSORFLOW":
return Optional.of(TENSORFLOW);
case "TENSORFLOW":
return Optional.of(TENSORFLOW);
default:
return Optional.empty();
}
......
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