Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import math
- oceniPoKorisnici={
- 'Lisa Rose': {'Catch Me If You Can': 3.0 , 'Snakes on a Plane': 3.5, 'Superman Returns': 3.5, 'You, Me and Dupree': 2.5, 'The Night Listener': 3.0, 'Snitch': 3.0},
- 'Gene Seymour': {'Lady in the Water': 3.0, 'Snakes on a Plane': 3.5, 'Just My Luck': 1.5, 'The Night Listener': 3.0,'You, Me and Dupree': 3.5},
- 'Michael Phillips': {'Catch Me If You Can': 2.5, 'Lady in the Water': 2.5,'Superman Returns': 3.5, 'The Night Listener': 4.0, 'Snitch': 2.0},
- 'Claudia Puig': {'Snakes on a Plane': 3.5, 'Just My Luck': 3.0,'The Night Listener': 4.5, 'Superman Returns': 4.0,'You, Me and Dupree': 2.5},
- 'Mick LaSalle': {'Lady in the Water': 3.0, 'Snakes on a Plane': 4.0,'Just My Luck': 2.0, 'Superman Returns': 3.0, 'You, Me and Dupree': 2.0},
- 'Jack Matthews': {'Catch Me If You Can': 4.5, 'Lady in the Water': 3.0, 'Snakes on a Plane': 4.0,'The Night Listener': 3.0, 'Superman Returns': 5.0, 'You, Me and Dupree': 3.5, 'Snitch': 4.5},
- 'Toby': {'Snakes on a Plane':4.5, 'Snitch': 5.0},
- 'Michelle Nichols': {'Just My Luck' : 1.0, 'The Night Listener': 4.5, 'You, Me and Dupree': 3.5, 'Catch Me If You Can': 2.5, 'Snakes on a Plane': 3.0},
- 'Gary Coleman': {'Lady in the Water': 1.0, 'Catch Me If You Can': 1.5, 'Superman Returns': 1.5, 'You, Me and Dupree': 2.0},
- 'Larry': {'Lady in the Water': 3.0, 'Just My Luck': 3.5, 'Snitch': 1.5, 'The Night Listener': 3.5}
- }
- def sim_distance(prefs,person1,person2):
- si={}
- for item in prefs[person1]:
- if item in prefs[person2]:
- si[item]=1
- if len(si)==0: return 0
- sum_of_squares=sum([pow(prefs[person1][item]-prefs[person2][item],2)
- for item in prefs[person1] if item in prefs[person2]])
- return round(1/(1+math.sqrt(sum_of_squares)),3)
- def sim_pearson(oceni,person1,person2):
- si={}
- for item in oceni[person1]:
- if item in oceni[person2]:
- si[item] = 1
- n = len(si)
- if n==0:
- return 0
- sum1=sum([oceni[person1][it] for it in si])
- sum2=sum([oceni[person2][it] for it in si])
- sum1Sq=sum([pow(oceni[person1][it],2) for it in si])
- sum2Sq=sum([pow(oceni[person2][it],2) for it in si])
- pSum=sum([oceni[person1][it]*oceni[person2][it] for it in si])
- num=pSum-(sum1*sum2/n)
- den=math.sqrt((sum1Sq-pow(sum1,2)/n)*(sum2Sq-pow(sum2,2)/n))
- if den==0: return 0
- r=num/den
- return round(r,3)
- def sim (oceni,person1, person2):
- numb=0
- for item in oceni[person1]:
- if item in oceni[person2]:
- numb+=1
- return numb
- def TabelaNaSlicniKorisnici(oceni):
- slicnosti={}
- for p1 in oceni:
- for p2 in oceni:
- slicnosti.setdefault(p1,{})
- slicnosti[p1][p2] = (sim_distance(oceni,p1,p2),sim_pearson(oceni,p1,p2),sim(oceni,p1,p2))
- return slicnosti
- if __name__ == "__main__":
- korisnik1=input()
- korisnik2=input()
- # korisnik1='Mick LaSalle'
- # korisnik2='Lisa Rose'
- # print oceniPoKorisnici
- tabela=TabelaNaSlicniKorisnici(oceniPoKorisnici)
- # print tabela
- print tabela[korisnik1][korisnik2]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement