Adding constructor for alternate matrix factory

ml4jarchive · Feb 20, 2020 · 3f7c3f0 · 3f7c3f0
1 parent cbd1b03
commit 3f7c3f0
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/java/org/ml4j/nn/quickstart/sessions/factories/QuickstartSessionFactory.java b/src/main/java/org/ml4j/nn/quickstart/sessions/factories/QuickstartSessionFactory.java
@@ -54,6 +54,13 @@ public QuickstartSessionFactory(boolean isTrainingContext) {
 		this(DEFAULT_MATRIX_FACTORY, createDefaultDirectedComponentsContext(DEFAULT_MATRIX_FACTORY, isTrainingContext));
 	}
 
+	/**
+	 * Quickstart session factory with all factories set to defaults except for matrix factory
+	 */
+	public QuickstartSessionFactory(MatrixFactory matrixFactory, boolean isTrainingContext) {
+		this(matrixFactory, createDefaultDirectedComponentsContext(DEFAULT_MATRIX_FACTORY, isTrainingContext));
+	}
+
 	/**
 	 * Quickstart session factory with all factories set to defaults, but with provided directedComponentsContext
 	 */