Skip to content

Commit ef0cae7

Browse files
committed
Remove model-specific alignment
1 parent 5fcd7fc commit ef0cae7

4 files changed

Lines changed: 0 additions & 175 deletions

File tree

README.md

Lines changed: 0 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -48,38 +48,6 @@ Some related projects:
4848
Let's [**Get Started!**](https://deepctr-doc.readthedocs.io/en/latest/Quick-Start.html)([Chinese
4949
Introduction](https://zhuanlan.zhihu.com/p/53231955)) and [welcome to join us!](./CONTRIBUTING.md)
5050

51-
## Consistent DeepFM defaults
52-
53-
DeepFM uses semantic-name-based portable initialization by default, so the
54-
normal Keras API starts from the same parameters as DeepCTR-Torch:
55-
56-
```python
57-
from deepctr.models import DeepFM
58-
model = DeepFM(
59-
linear_feature_columns,
60-
dnn_feature_columns,
61-
seed=2026,
62-
)
63-
model.compile("adam", "binary_crossentropy", metrics=["AUC"])
64-
model.fit(x, y, batch_size=1024, epochs=1)
65-
```
66-
67-
Portable initialization is currently implemented for DeepFM. Model architecture
68-
and feature-column ordering must match when comparing independent TensorFlow and
69-
PyTorch runs.
70-
71-
For portable inference, transfer the trained semantic parameters without
72-
depending on backend variable names:
73-
74-
```python
75-
from deepctr.initializers import export_deepfm_parameters
76-
77-
portable_parameters = export_deepfm_parameters(model)
78-
```
79-
80-
Pass the resulting dictionary to
81-
`deepctr_torch.initializers.load_deepfm_parameters(torch_model, portable_parameters)`.
82-
8351
## Models List
8452

8553
| Model | Paper |

deepctr/initializers.py

Lines changed: 0 additions & 102 deletions
This file was deleted.

deepctr/models/deepfm.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
from tensorflow.keras.layers import Dense
1515

1616
from ..feature_column import build_input_features, get_linear_logit, DEFAULT_GROUP_NAME, input_from_feature_columns
17-
from ..initializers import initialize_deepfm_parameters
1817
from ..layers.core import PredictionLayer, DNN
1918
from ..layers.interaction import FM
2019
from ..layers.utils import concat_func, add_func, combined_dnn_input
@@ -63,5 +62,4 @@ def DeepFM(linear_feature_columns, dnn_feature_columns, fm_group=(DEFAULT_GROUP_
6362

6463
output = PredictionLayer(task)(final_logit)
6564
model = Model(inputs=inputs_list, outputs=output)
66-
initialize_deepfm_parameters(model, seed=seed)
6765
return model

tests/models/CrossFrameworkTraining_test.py

Lines changed: 0 additions & 39 deletions
This file was deleted.

0 commit comments

Comments
 (0)