Skip to content

changing integer inputs to float #4

@ibozkurt79

Description

@ibozkurt79

Keras expects float in model training so model definition should be;

train model

history = train_model(GMF_model, 'adam', BATCH_SIZE, EPOCHS, VAL_SPLIT,
inputs=[df_train.userId.values.astype(np.float32), df_train.movieId.values.astype(np.float32)],
outputs=df_train.rating.values)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions