CORA create objective function

cora_sklearn_objective.py

def objectiveSKLearn():
    
    def objective(params):

        dim = params['embeddingDimension']
        thirdWeight = params['thirdWeight']
        fourthWeight = params['fourthWeight']
        normalizationStrength = params['normalizationStrength']

        # Create embeddings
        query = """CALL gds.fastRP.write(
                    'cora',
                    {
                        embeddingDimension: %d,
                        iterationWeights: [0.0, 0.0, %f, %f],
                        normalizationStrength: %f,
                        randomSeed: 42,
                        writeProperty: 'fastrp_embedding'
                    }
                )
        """ % (dim, thirdWeight, fourthWeight, normalizationStrength)

        conn.query(query)

        return modeler(params)
    
    return objective