tffm
tffm copied to clipboard
TensorFlow implementation of an arbitrary order Factorization Machine
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...
@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.
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...
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...
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,...