From b7f1f6e2130e1156295543542298265b8f5b6e16 Mon Sep 17 00:00:00 2001
From: Owen Green <gungwho@gmail.com>
Date: Mon, 6 Jul 2020 18:33:38 +0100
Subject: [PATCH] Add FluidMLPRegressor

---
 .../Classes/FluidMLPRegressor.sc              | 35 +++++++++++++++++++
 src/FluidManipulation/FluidManipulation.cpp   |  2 ++
 2 files changed, 37 insertions(+)
 create mode 100644 release-packaging/Classes/FluidMLPRegressor.sc

diff --git a/release-packaging/Classes/FluidMLPRegressor.sc b/release-packaging/Classes/FluidMLPRegressor.sc
new file mode 100644
index 0000000..33443aa
--- /dev/null
+++ b/release-packaging/Classes/FluidMLPRegressor.sc
@@ -0,0 +1,35 @@
+FluidMLPRegressor : FluidManipulationClient {
+
+  classvar <identity = 0;
+  classvar <sigmoid  =  1;
+  classvar <relu = 2;
+  classvar <tanh = 3;
+
+	*new {|server, hidden = #[3,3] , activation = 0, maxIter = 100, learnRate = 0.0001, momentum = 0.9, batchSize = 50|
+		var uid = UniqueID.next;
+		^super.new(server,*([hidden.size]++hidden++activation++maxIter++learnRate++
+			momentum++batchSize++uid))!?{|inst|inst.init(uid);inst}
+	}
+
+	init {|uid|
+		id = uid;
+	}
+
+	fit{|sourceDataset, targetDataset, action|
+	   this.prSendMsg(\fit,
+			[sourceDataset.asSymbol, targetDataset.asSymbol],
+			action,numbers(FluidMessageResponse,_,1,_)
+		);
+	}
+
+	predict{ |sourceDataset, targetDataset, layer, action|
+		this.prSendMsg(\predict,
+			[sourceDataset.asSymbol, targetDataset.asSymbol,layer],
+			action);
+	}
+
+	predictPoint { |sourceBuffer, targetBuffer, layer action|
+		this.prSendMsg(\predictPoint,
+      [sourceBuffer.asUGenInput, targetBuffer.asUGenInput, layer], action);
+	}
+}
diff --git a/src/FluidManipulation/FluidManipulation.cpp b/src/FluidManipulation/FluidManipulation.cpp
index 4df8b6f..a0c0738 100644
--- a/src/FluidManipulation/FluidManipulation.cpp
+++ b/src/FluidManipulation/FluidManipulation.cpp
@@ -12,6 +12,7 @@
 #include <clients/nrt/StandardizeClient.hpp>
 #include <clients/nrt/PCAClient.hpp>
 #include <clients/nrt/MDSClient.hpp>
+#include <clients/nrt/MLPRegressorClient.hpp>
 #include <clients/rt/AudioTransportClient.hpp>
 #include <clients/rt/FluidDataSetWr.hpp>
 #include <FluidSCWrapper.hpp>
@@ -36,4 +37,5 @@ PluginLoad(FluidSTFTUGen)
   makeSCWrapper<RTAudioTransportClient>("FluidAudioTransport",ft);
   makeSCWrapper<NRTThreadedAudioTransportClient>("FluidBufAudioTransport",ft);
   makeSCWrapper<NRTThreadedDataSetWriter>("FluidDataSetWr", ft);
+  makeSCWrapper<NRTThreadedMLPRegressorClient>("FluidMLPRegressor",ft);
 }