Namonay/ft_linear_regression
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add: better input handling

Browse Source
This commit is contained in:
Namonay
2025-05-04 22:09:31 +02:00
parent a94e933e75
commit afb04c043a
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
estimate.py
View File

@@ -7,13 +7,20 @@ class Estimate:
self.theta1 = 0 self.theta1 = 0
self.get_thetas() self.get_thetas()
def get_thetas(self): def get_thetas(self):
try:
with open(self.path, 'r') as file: with open(self.path, 'r') as file:
self.data = pd.read_csv(file) self.data = pd.read_csv(file)
self.theta0 = self.data["theta0"].iloc[0] self.theta0 = self.data["theta0"].iloc[0]
self.theta1 = self.data["theta1"].iloc[0] self.theta1 = self.data["theta1"].iloc[0]
except:
print("! Warning, no data has been found")
def estimate_price(self, mileage): def estimate_price(self, mileage):
return self.theta0 + (self.theta1 * mileage) return self.theta0 + (self.theta1 * mileage)
estimate = Estimate() estimate = Estimate()
input = int(input("Please enter the mileage of your car: ")) try:
print("The estimated price for a", input, "kilometers car is :", estimate.estimate_price(input)) input = int(input("Please enter the mileage of your car: "))
except:
print("! Wrongs args format")
else:
print("The estimated price for a", input, "kilometers car is :", estimate.estimate_price(input))