tffm icon indicating copy to clipboard operation
tffm copied to clipboard

TensorFlow implementation of an arbitrary order Factorization Machine

Results 19 tffm issues
Sort by recently updated
recently updated
newest added

When running tffm on tensorflow 1.7, the following warning appears: `WARNING:tensorflow:Variable += will be deprecated. Use variable.assign_add if you want assignment to the variable value or 'x = x +...

I need to run multiple `TFFMRegressor` objects in `joblib` `Parallel`. To do so, I passed the following parameter: ``` session_config=tf.ConfigProto(intra_op_parallelism_threads=1, inter_op_parallelism_threads=1, allow_soft_placement=True, device_count = {'CPU': 1, 'GPU': 0}) ``` However,...

Currently the error is "NameError: name 'sklearn' is not defined" which is not the intended one as sklearn is not in scope. So adding a import statement.

I've been looking at Spark implementations of Factorization Machines. I found that none of the existing open source implementations scale to a dataset with millions of features and hundreds of...

help wanted

@geffy frist thank you for provide us this good tool,but in fact sometime there some question of multi-class .Could you give some advises.

enhancement

Need to compare tffm and libfm in terms of speed and quality.

help wanted

Hi, I just wondered how FM can be parallelized effectively between multiple GPUs. I'm a bit familiar with TF and not really with FMs. If you provide me with ideas...

enhancement
help wanted

Hi, Would it be possible to automatically save the "best" model run in terms of global loss? Maybe in a way similar to how https://keras.io/callbacks/#modelcheckpoint works? I had different runs...

enhancement

Hi, thank you for making this great package! I just noticed this example for `get_shorter_decompositions` function in `util.py`: ``` Example ------- decompositions, counts = get_shorter_decompositions([1, 2, 3]) decompositions == [(1,...