FilesExpand file tree

 main

/

SELFRec.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

25 lines (21 loc) · 1.1 KB

 main

/

SELFRec.py

Copy path

Top

File metadata and controls

• Code
• Blame

25 lines (21 loc) · 1.1 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

from data.loader import FileIO

class SELFRec(object):

def __init__(self, config):

self.social_data = []

self.feature_data = []

self.config = config

self.training_data = FileIO.load_data_set(config['training.set'], config['model']['type'])

self.test_data = FileIO.load_data_set(config['test.set'], config['model']['type'])

self.kwargs = {}

if config.contain('social.data'):

social_data = FileIO.load_social_data(self.config['social.data'])

self.kwargs['social.data'] = social_data

# if config.contains('feature.data'):

# self.social_data = FileIO.loadFeature(config,self.config['feature.data'])

print('Reading data and preprocessing...')

def execute(self):

# import the model module

import_str = f"from model.{self.config['model']['type']}.{self.config['model']['name']} import {self.config['model']['name']}"

exec(import_str)

recommender = f"{self.config['model']['name']}(self.config,self.training_data,self.test_data,**self.kwargs)"

eval(recommender).execute()