taegeonum commented on a change in pull request #203: [NEMO-360] Implementing
an 'XGBoostPolicy'
URL: https://github.com/apache/incubator-nemo/pull/203#discussion_r269933413
##########
File path:
compiler/optimizer/src/main/java/org/apache/nemo/compiler/optimizer/NemoOptimizer.java
##########
@@ -113,6 +124,23 @@ public IRDAG optimizeAtRunTime(final IRDAG dag, final
Message message) {
return optimizationPolicy.runRunTimeOptimizations(dag, message);
}
+ /**
+ * Operations to be done prior to the Compile-Time Optimizations.
+ * @param dag the DAG to process.
+ * @param policy the optimization policy to optimize the DAG with.
+ */
+ private void beforeCompileTimeOptimization(final IRDAG dag, final Policy
policy) {
+ if (policy instanceof XGBoostPolicy) {
+ clientRPC.send(ControlMessage.DriverToClientMessage.newBuilder()
+ .setType(ControlMessage.DriverToClientMessageType.LaunchOptimization)
+ .setOptimizationType("xgboost")
+ .setDataCollected(ControlMessage.DataCollectMessage.newBuilder()
+ .setData(dag.irDAGSummary() + this.environmentTypeStr)
+ .build())
+ .build());
+ }
+ }
Review comment:
Could you please explain which implementation issues do you have?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
