File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
lib/smurff-cpp/SmurffCpp/Predict Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -238,9 +238,9 @@ std::shared_ptr<Model> PredictSession::restoreModel(const std::shared_ptr<StepFi
238
238
return model;
239
239
}
240
240
241
- std::shared_ptr<Model> PredictSession::restoreModel (int i)
241
+ std::shared_ptr<Model> PredictSession::restoreModel (int i, int skip_mode )
242
242
{
243
- return restoreModel (m_stepfiles.at (i));
243
+ return restoreModel (m_stepfiles.at (i), skip_mode );
244
244
}
245
245
246
246
// predict one element
Original file line number Diff line number Diff line change @@ -40,7 +40,7 @@ class PredictSession : public ISession
40
40
41
41
private:
42
42
std::shared_ptr<Model> restoreModel (const std::shared_ptr<StepFile> &, int skip_mode = -1 );
43
- std::shared_ptr<Model> restoreModel (int i);
43
+ std::shared_ptr<Model> restoreModel (int i, int skip_mode = - 1 );
44
44
45
45
public:
46
46
int getNumSteps () const { return m_stepfiles.size (); }
You can’t perform that action at this time.
0 commit comments