Merge pull request #47 from akleeman/default_template_predict

akleeman · web-flow · commit f62c5335f981 · 2018-08-09T15:44:41.000-07:00
Add a default template parameter for predict(feature).
diff --git a/albatross/core/model.h b/albatross/core/model.h
@@ -97,7 +97,7 @@ class RegressionModel : public ParameterHandlingMixin {
     return predict(features, detail::PredictTypeIdentity<PredictType>());
   }
 
-  template <typename PredictType>
+  template <typename PredictType = JointDistribution>
   PredictType predict(const FeatureType &feature) const {
     std::vector<FeatureType> features = {feature};
     return predict<PredictType>(features);

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ class RegressionModel : public ParameterHandlingMixin {`
`97`	`97`	`return predict(features, detail::PredictTypeIdentity<PredictType>());`
`98`	`98`	`}`
`99`	`99`
`100`		`- template <typename PredictType>`
	`100`	`+ template <typename PredictType = JointDistribution>`
`101`	`101`	`PredictType predict(const FeatureType &feature) const {`
`102`	`102`	`std::vector<FeatureType> features = {feature};`
`103`	`103`	`return predict<PredictType>(features);`