com
.
intel
.
analytics
.
zoo
.
examples
recommendation
package
recommendation
Visibility
Public
All
Type Members
case class
Item
(
itemId:
Int
,
title:
String
,
genres:
String
)
extends
Product
with
Serializable
case class
NeuralCFParams
(
inputDir:
String
=
"./data/ml-1m"
,
batchSize:
Int
=
2800
,
nEpochs:
Int
=
10
,
learningRate:
Double
=
1e-3
,
learningRateDecay:
Double
=
1e-6
)
extends
Product
with
Serializable
case class
Rating
(
userId:
Int
,
itemId:
Int
,
label:
Int
)
extends
Product
with
Serializable
case class
Record
(
age:
Int
,
workclass:
String
,
fnlwgt:
Int
,
education:
String
,
education_num:
Int
,
marital_status:
String
,
occupation:
String
,
relationship:
String
,
race:
String
,
gender:
String
,
capital_gain:
Int
,
capital_loss:
Int
,
hours_per_week:
Int
,
native_country:
String
,
income_bracket:
String
)
extends
Product
with
Serializable
case class
User
(
userId:
Int
,
gender:
String
,
age:
Int
,
occupation:
Int
)
extends
Product
with
Serializable
case class
WNDParams
(
dataset:
String
=
"ml-1m"
,
modelType:
String
=
"wide_n_deep"
,
inputDir:
String
=
"./data/ml-1m/"
,
onSpark:
Boolean
=
true
,
batchSize:
Int
=
2048
,
maxEpoch:
Int
=
10
,
logDir:
Option
[
String
] =
None
)
extends
Product
with
Serializable
Value Members
object
CensusWideAndDeep
object
Ml1mWideAndDeep
object
NeuralCFexample
object
WideAndDeepExample
Ungrouped