forked from turi-code/python-libffm
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbasic.py
More file actions
23 lines (19 loc) · 638 Bytes
/
basic.py
File metadata and controls
23 lines (19 loc) · 638 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import graphlab as gl
import ffm
from convert import read_libffm_file
# Create SFrames from example text files provided with libffm
trainfile = 'lib/bigdata.tr.txt'
validfile = 'lib/bigdata.te.txt'
train = read_libffm_file(trainfile)
valid = read_libffm_file(validfile)
train['y'] = train['y'].astype(int)
del train['features.0']
valid = valid[train.column_names()]
train.save('examples/small.tr.sframe')
valid.save('examples/small.te.sframe')
features = [c for c in train.column_names() if c != 'y']
# Train a model
m = ffm.FFM()
m.fit(train, valid, target='y', features=features, nr_iters=15)
# yhat = m.predict(valid)
# print yhat