File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -3068,15 +3068,16 @@ TEST_CASE("PredictSession/Features/2"
3068
3068
std::shared_ptr<ISession> session = SessionFactory::create_session (config);
3069
3069
session->run ();
3070
3070
3071
- PredictSession predict_session (session->getRootFile ());
3072
- auto in_matrix_predictions = predict_session .predict (config.getTest ())->m_predictions ;
3071
+ PredictSession predict_session_in (session->getRootFile ());
3072
+ auto in_matrix_predictions = predict_session_in .predict (config.getTest ())->m_predictions ;
3073
3073
3074
+ PredictSession predict_session_out (session->getRootFile ());
3074
3075
auto sideInfoMatrix = matrix_utils::sparse_to_eigen (*rowSideInfoConfig->getSideInfo ());
3075
3076
int d = config.getTrain ()->getDims ()[0 ];
3076
3077
for (int r = 0 ; r < d; r++)
3077
3078
{
3078
3079
auto feat = sideInfoMatrix.row (r).transpose ();
3079
- auto out_of_matrix_predictions = predict_session .predict (0 , feat);
3080
+ auto out_of_matrix_predictions = predict_session_out .predict (0 , feat);
3080
3081
// Eigen::VectorXd out_of_matrix_averages = out_of_matrix_predictions->colwise().mean();
3081
3082
3082
3083
#undef DEBUG_OOM_PREDICT
You can’t perform that action at this time.
0 commit comments