You are on page 1of 2

import numpy as np

import random

class base():
def __init__(self,nbr,dim):
self.nbr=nbr
self.dim=dim

def valeurs(self):
self.label=np.zeros(self.nbr)
self.data=np.zeros((self.nbr,self.dim))

for i in range(self.nbr//2):
self.label[i]=0
for j in range(self.dim):
self.data[i][j]=random.randint(100,200)

for i in range(self.nbr//2,self.nbr):
self.label[i]=1
for j in range(self.dim):
self.data[i][j]=random.randint(0,100)

def display(self):
choix=input('what do u want to print?: ')
if choix=='label':
print(self.label)
if choix=='data':
print(self.data)

class classif():
def __init__(self):
pass

def classer(self,test,A):
s=0

d=np.zeros((A.nbr))

for i in range(A.nbr):
for j in range(A.dim):
s+=(test[j]-A.data[i][j])**2

d[i]=np.sqrt(s)
Min=np.argmin(d)
print(d)
return A.label[Min]

A=base(20,40)
C1=A.valeurs()
C2=A.display()

B=classif()

test=np.zeros(40)
for i in range(40):
test[i]=random.randint(0,100)

B.classer(test,A)

You might also like