Skip to content

Commit ccc164f

Browse files
committed
FIX: do not mix in and out PredictSessions
1 parent 73189da commit ccc164f

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

lib/smurff-cpp/Tests/TestsSmurff.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3068,15 +3068,16 @@ TEST_CASE("PredictSession/Features/2"
30683068
std::shared_ptr<ISession> session = SessionFactory::create_session(config);
30693069
session->run();
30703070

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;
30733073

3074+
PredictSession predict_session_out(session->getRootFile());
30743075
auto sideInfoMatrix = matrix_utils::sparse_to_eigen(*rowSideInfoConfig->getSideInfo());
30753076
int d = config.getTrain()->getDims()[0];
30763077
for (int r = 0; r < d; r++)
30773078
{
30783079
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);
30803081
//Eigen::VectorXd out_of_matrix_averages = out_of_matrix_predictions->colwise().mean();
30813082

30823083
#undef DEBUG_OOM_PREDICT

0 commit comments

Comments
 (0)