()
| 42 | |
| 43 | |
| 44 | def test_DIN(): |
| 45 | model_name = "DIN" |
| 46 | |
| 47 | x, y, feature_columns, behavior_feature_list = get_xy_fd() |
| 48 | model = DIN(feature_columns, behavior_feature_list, dnn_dropout=0.5, device=get_device()) |
| 49 | |
| 50 | check_model(model, model_name, x, y) # only have 3 train data so we set validation ratio at 0 |
| 51 | |
| 52 | |
| 53 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected