| package io.prediction.examples.java.recommendations.tutorial4; |
| |
| import java.io.Serializable; |
| import java.util.Map; |
| import org.apache.commons.math3.linear.RealVector; |
| |
| public class FeatureBasedModel implements Serializable { |
| // Feature value is always between -1 and 1. |
| public final Map<Integer, RealVector> userFeatures; |
| public final Map<Integer, Integer> userActions; |
| public final Map<Integer, RealVector> itemFeatures; |
| |
| public FeatureBasedModel( |
| Map<Integer, RealVector> userFeatures, |
| Map<Integer, Integer> userActions, |
| Map<Integer, RealVector> itemFeatures) { |
| this.userFeatures = userFeatures; |
| this.userActions = userActions; |
| this.itemFeatures = itemFeatures; |
| } |
| } |
| |