🌐 AI搜索 & 代理 主页
Skip to content

MVP goals #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
levkk merged 17 commits into from
Apr 14, 2022
Merged

MVP goals #1

Changes from 1 commit
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
92b80f2
MVP goals
Apr 12, 2022
f18f276
Use unittest as the test running harness
Apr 12, 2022
3c66272
remove validate because validation has a different meaning in ML, and…
Apr 12, 2022
958cfba
keep model in memory to avoid going to disk
Apr 12, 2022
14b1f61
use bytea directly for pl/python rather than hex/text conversion
Apr 12, 2022
829b62e
add a draft schema to support snapshots and multiple training runs fo…
Apr 12, 2022
9907aaa
sketch out the regression model training cycle
Apr 13, 2022
b50f000
break it down into model classes
Apr 13, 2022
89b467d
add categoricals
Apr 14, 2022
d9d6727
Update pgml/tests/test_train.py
montanalow Apr 14, 2022
dfb57c6
fix categorical test
Apr 14, 2022
56e033d
Merge branch 'montana/readme' of github.com:postgresml/postgresml int…
Apr 14, 2022
a1ef909
docs
Apr 14, 2022
c2de3d8
make test that "works"
Apr 14, 2022
ffedbc5
Update pgml/pgml/model.py
montanalow Apr 14, 2022
aa44f94
remove parens around ifs
Apr 14, 2022
4ca1a5f
Merge branch 'montana/readme' of github.com:postgresml/postgresml int…
Apr 14, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Use unittest as the test running harness
  • Loading branch information
Montana Low committed Apr 12, 2022
commit f18f276be63af5fef7601c2fb3c8e908f65815fc
31 changes: 17 additions & 14 deletions pgml/tests/test_train.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
import unittest
from pgml.train import train


Expand All @@ -14,18 +15,20 @@ def fetch(self, n):
return self._values


def test_train():
it = PlPyIterator(
[
{
"value": 5,
"weight": 5,
},
{
"value": 34,
"weight": 5,
},
]
)
class TestTrain(unittest.TestCase):
def test_train(self):
it = PlPyIterator(
[
{
"value": 5,
"weight": 5,
},
{
"value": 34,
"weight": 5,
},
]
)

train(it, y_column="weight", name="test", save=False)
train(it, y_column="weight", name="test", save=False)
self.assertTrue(True)